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

1 篇文章 0 订阅
1 篇文章 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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值