截止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"