截止2018.2.21日,tensorflow1.5並不能支持最新版本的cuda9.1
- 因此,想要安裝gpu版本需安裝tensorflow1.4版本(支持cuda8)
sudo python3 -m pip install tensorflow-gpu==1.4
- CUDA8目前有兩個版本,GA2(發佈於2017年2月)和GA1(發佈於2016年9月)
我選的是GA2,runfile(local) --- deb(local)文件在安裝時會無法選擇版本,在安裝好配置文件後,會默認安裝最新版本
安裝過程中會提示是否需要安裝驅動(我選的文件默認安裝驅動375.26),你可以打開自己的系統設置-軟件更新-附加驅動,裏面查看自己是否有NVIDIA的驅動(cuda8支持365以上版本)。如果你已有顯卡驅動,就不要再重複安裝別的版本驅動。
- cudnn對應cuda8也有兩個版本,要選擇cudnn v6.0,不然在import tensorflow時會出現(ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory)錯誤
安裝時要安裝cuDNN v6.0 Runtime 和 Developer 兩個文件,(code samples可選安裝)
- 在使用過程中,如果出現ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory,需要
在環境變量中加入
二者缺一不可(這裏的路徑是安裝時候的默認路徑,具體路徑因人而異)export PATH="$PATH:/usr/local/cuda-8.0/bin" export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"