Ubuntu16.04下GTX970配置機器學習環境
阿新 • • 發佈：2019-01-05
首先下載以下三個軟體
1、安裝顯示卡
ctrl+alt+F1 進入命令列模式，關掉方式：sudo /etc/init.d/lightdm stop，然後開始安裝顯示卡驅動：
直接運雄驅動的*.run檔案即可。中間會有提示，選yes沒有害處，可能會提示解除安裝之前的驅動，選yes即可。最後提示安裝成功，會回到影象桌面，手動重啟一下。
但是好幾次都沒有成功，最後找到了一個簡單的方法，那就是在ubuntu的軟體管理裡面，找到額外驅動那一項，會看到你的顯示卡情況，可以直接在這裡安裝驅動。2、安裝cuda
同樣進入命令列，關掉X server服務，執行run檔案。在選擇是否安裝驅動時選擇否，其他的選擇Yes。
結束之後新增環境變數
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH
3、安裝cuDNN這個是把一些檔案複製到指定位置即可：
tar xvzf cudnn-7.0-linux-x64-v4-prod.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
4、新增環境變數
否則TF不知道CUDA的東西在哪裡，方法：在/etc/environment檔案中直接新增下面路徑：
:/usr/local/cuda/lib64:/usr/local/cuda
至此差不多就可以跑DNN了!