【指南】Deepin 15.5安装tensorflow-gpu 1.4- Community

【指南】Deepin 15.5安装tensorflow-gpu 1.4

Apps Section 1501 views · 4 replies ·

javanli

deepin

2018-06-02 21:26

Author

之前看了几篇安装tensorflow-gpu 1.5/1.8 + cuda9.0的帖子，实际操作比较复杂，而且直接把我系统搞挂了...

如果不是很需要tensorflow最新版本的话，安装tensorflow-gpu 1.4 + cuda8.0会比较简单。
本文不适合更高版本的tensorflow-gpu安装。

1. 安装NVIDIA驱动
直接打开deepin自带的驱动管理器安装即可。

2. 安装CUDA 8.0
一行命令即可
sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler

3. 安装cudnn
去官网https://developer.nvidia.com/cudnn下载Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0(https://developer.nvidia.com/rdp/cudnn-archive#a-collapse6-8)
即依赖cuda8.0的cudnn6.0
下载下来是个压缩包，解压到主目录(~)即可。
然后需要配置LD_LIBRARY_PATH，编辑~/.bashrc，在最后一行加上export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/xxxxx/cuda/lib64"
把xxxxx换成你的账户，/home/xxxxx/cuda就是刚刚解压出的文件夹，你也可以放到别的地方。

4. 重启系统，使驱动生效

5. 安装tensorflow-gpu
sudo pip3 install tensorflow-gpu==1.4
注意这里必须是1.4，对应CUDA8.0。
(deepin自带python和python3，对应2.7和3.5版本。sudo apt install python-pip python3-pip可以安装pip和pip3命令，这里用2.7还是3.5随意即可)

6. 验证安装
$ python3
运行一个简短的程序
# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

Reply Like 0 Favorite View the author

All Replies