准备:
Ubuntu16.04
Anaconda3-4.2.0-Linux-x86.sh
cuda_8.0.61_375.26_linux.run
cudnn-8.0-linux-x64-v6.0.solitairetheme8
tensorflow_gpu-1.3.0-cp36-cp36m-manylinux1_x86_64.whl
!!注意,TensorFlow-GPU、Cuda、Cudnn三个版本一定要对的上才能安装成功!!
- 先安装anaconda:进入拷贝出来的目录(一般放在Home下就好)
sudo bash Anaconda3-4.2.0-Linux-x86.sh
安装好以后测试一下,如果出现
说明现在默认的python编译器是anaconda下的python编译器。
- 安装cuda
首先进入系统设置-》软件更新-》附加驱动,找到选择NVIDIA binary driver专有,应用更改,直接它给安装好了N卡驱动,省去自己卸载安装显卡驱动。
切换以后,返回来安装cuda
sudo bash cuda_8.0.61_375.26_linux.run
按空格键把那些协议给跳完,然后回车、输入yes等进行安装。
- 安装cudnn
把这个文件cudnn-8.0-linux-x64-v6.0.solitairetheme8文件格式后缀改成cudnn-8.0-linux-x64-v6.0.tgz,
解压:
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
拷贝到cuda下,修改权限
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*
- 安装TensorFlow
pip install tensorflow_gpu-1.3.0-cp36-cp36m-manylinux1_x86_64.whl
在运行import tensorflow时候,可能出现:
Import Error:libcusolver.so.6.0: cannot open shared object file:No Such file……
运行
sudo ldconfig /usr/local/cuda/lib64
如果运行上述语句出错, *.so不是符号连接
sudo -sf /usr/local/cuda-8.0/lib64/libcudnn.so.6.0.21 /usr/local/cuda-8.0/lib64/libcudnn.so.6
再运行上面的ldconfig