Deepin 15.7安装tensorflow-gpu的一些经验
Tofloor
poster avatar
wangjj1991
deepin
2018-10-19 22:10
Author
本帖最后由 wangjj1991 于 2018-10-19 14:12 编辑

之前看了几篇安装tensorflow-gpu的帖子,然后在重装几次系统的基础上分享下个人成功安装tensorflow-gpu==1.9的经验。主要参考了帖子
1.【指南】Deepin 15.5安装 tensorflow-gpu 1.4 ,https://bbs.deepin.org/post/157491#tensorflow
2. deepin15.7安装tensorflow+cuda9.0+cudnn7, https://www.jianshu.com/p/caa02259e51d

我的系统是
deepin==15.7,
显卡GTX965M,

CPU i7-4710MQ.


1. 安装NVIDIA驱动
打开驱动管理器,选择prime方案,安装驱动,装完之后驱动版本我的是390.67
可以apt install nvidia-smi 或者在nvidia驱动管理器查看
2. 安装CUDA 9.1
sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler
可以直接安装CUDA 9.1,这边参考的第一个帖子里面安装的是8.0,我这边安装的是CUDA9.1
3. 安装cudnn
去官网https://developer.nvidia.com/cudnn下载 Download cuDNN v7.1.3 (April 17, 2018), for CUDA 9.1
下载下来是个压缩包,解压到主目录(~)即可。
然后需要配置LD_LIBRARY_PATH,编辑~/.bashrc,在最后一行加上export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/xxxxx/cuda/lib64"
把xxxxx换成你的账户,/home/xxxxx/cuda就是刚刚解压出的文件夹,你也可以放到别的地方。4. source ~/.bashrc,是更改生效
5. 安装tensorflow-gpu
这边我使用的Miniconda,
conda install tensorlflow-gpu==1.9
6. 验证安装
$ python3
运行一个简短的程序
# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

-----
PS1. 第一次参考的帖子2,也是直接使用Prime方案安装的N卡驱动,之后安装CUDA 9.0 和cuDNN7.0, tensorflow-gpu==1.7
PS2. Minicoda 会自己编译一个CUDA,cuDNN的库,安装的tensorflow会报 runtime 不符合的错误,可以试试更换gpu的版本号试试
PS3. CUDA要和nvidia drive的版本要对应。


Reply Favorite View the author
All Replies
avatar
aida
deepin
2018-10-19 23:15
#1
支持原创! 加油!
Reply View the author
avatar
177******80
deepin
2018-10-24 03:18
#2
本帖最后由 sksweet 于 2018-10-23 19:27 编辑

新手问一下,楼主,你注意过,程序运行的时候,cpu是多少度呀,我运行一个tensorflow的hello world,跑60000份数据集,才启动不到30秒,cpu就近100度了,我的系统是manjaor版本的Deepin.用的是tensorflow是tensorflow-GPU版本。这温度也太吓人了。做机器学习的时候,都是哪么高的温度吗??差点把直接把笔记本给烧了。。。

Reply View the author