軟件安裝

為了實現(xiàn)分布式,搞了一天的環(huán)境。之前安裝的cuda8.0,但是要訓練的keras-yolo3需要的tensorflow是1.6的版本,而這個版本對應的cuda是9.0版本。
本來想實現(xiàn)8.0,9.0共存,誰知一直出現(xiàn)類似libcudnn 9.0.so的錯誤。其實這里我應該是踩了坑,按照網上一些教程設置好后,其實應該將電腦重啟的。我卻選擇了卸掉cuda8.0,只裝了9.0.
這里記錄一下本人經常記不住的命令:
sudo -H pip3 list
sudo gedit /etc/profile    打開配置文件
sudo ln -s /usr/local/cuda /usr/local/cuda-9.0 掛起cuda
sudo pip3 install tensorflow-gpu==1.6.0    tensorflow安裝
sudo pip3 install keras    keras安裝

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容