Tensorflow-gpu安装过程记录

1. 先安装cuda。 下载cuda_8.0.44_linux.run,安装,安装路径/usr/local/cuda-8.0
2. 安装cudnn. 下载cudnn-8.0-linux-x64-v6.0.tgz,文件解压缩到/home/weizhong/software/cuda,把/home/weizhong/software/cuda/includ目录下的cudnn.h复制到/usr/local/cuda-8.0/include
3. 安装anaconda. 下载Anaconda3-5.0.1-Linux-x86_64.s4. 创建conda环境. conda create -n tensorflow python=3.5.  此处因为tensorflow 1.4版本只支持到python3.5
5. 卸载已有的tensorflow: pip uninstall tensorflow-gpu
   pip uninstall protobuf
6. source activate tensorflow, 在此conda环境下, sudo pip3 install tensorflow-gpu==1.4
注:
1.到写博文的时候,cuda最新版本是9.1,tensorflow的最新版本是1.5,但在我的12年出厂的联想电脑(NVIDIA GF630M)上tensorflow1.5对cuda9.1不支持,安装不成功多次后,我不得不降低cuda版本到8.0,对应的tensorflow的版本降到1.4. 然后安装成功。
2. 安装cuda时,有一个问题很重要,如果显示
You are attempting to install on an unsupported configuration.”
这样的信息,在接下来的

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 xxx xx?
选项中,输入“n” 表示不安装,否则安装不成功,我在此反复多次,最后在autocyz的博文帮助下才成功。感谢autocyz,参见:ubuntu16.04安装 cuda7.5 (网址:http://blog.csdn.net/autocyz/article/details/51841157)
3. 安装cuda和cudnn要把它们添加到环境变量,可查阅度娘。我是参考gtx_viewer的博文“Nvidia CUDA Toolkit的安装(cuda)”(见:https://www.cnblogs.com/jinggege/p/5766146.html)
sudo gedit /etc/profile
然后,在打开的文件最后添加(此处cuda-8.0是我cuda的安装路径,/home/weizhong/software/cuda表示cudnn解压缩的路径):
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/home/weizhong/software/cuda/lib64:$LD_LIBRARY_PATH
再添加lib库路径:
sudo vim /etc/ld.so.conf.d/cuda.conf
内容如下:
/usr/local/cuda-8.0/lib64
/home/weizhong/software/cuda/lib64
/cudnn_dir/lib64
/lib
/usr/lib
/usr/lib32
4. 创建conda环境很重要,能省不少麻烦事,然后在创建的环境下安装tensorflow

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值