1. 程式人生 > >機器學習ubuntu16+1080Ti+cuda+cudnn+tensorflow最簡單的方法

機器學習ubuntu16+1080Ti+cuda+cudnn+tensorflow最簡單的方法

1080Ti驅動,親測不需關閉自身顯示卡驅動

sudo apt-get update && sudo apt-get install nvidia-381

 

安裝cuda+cudnn+tensorflow

直接pip install tensorflow-GPU==1.3.0

這個版本的tensorflow會自動安裝cuda和cudnn

剛才測試了

用canda裝tensorflow-GPU可能出現無法使用CUDa的問題,需要用自身的pip

多次測試得到的懶人經驗!