Ubuntu16.04 + Anaconda3 + TensorFlow1.3 + cuda8.0 + cudnn8.0-v6.0

准备:

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三个版本一定要对的上才能安装成功!!

 

  1. 先安装anaconda:进入拷贝出来的目录(一般放在Home下就好)

sudo bash Anaconda3-4.2.0-Linux-x86.sh

安装好以后测试一下,如果出现

 

说明现在默认的python编译器是anaconda下的python编译器。

  1. 安装cuda

首先进入系统设置-》软件更新-》附加驱动,找到选择NVIDIA binary driver专有,应用更改,直接它给安装好了N卡驱动,省去自己卸载安装显卡驱动。

       切换以后,返回来安装cuda

       sudo bash cuda_8.0.61_375.26_linux.run

       按空格键把那些协议给跳完,然后回车、输入yes等进行安装。

  1. 安装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*
  1. 安装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

你可能感兴趣的:(OpenCV,TensorFlow,GPU,CUDA,CuDNN)