40、Tensorflow:在centos7下安装tensorflow-gpu 1.8版本教程
转自:here
tensorflow的安装分为cpu版本的安装和gpu版本的安装.
我的系统环境是 centos7 + GPU(24G显存)+python2.7
nvidia-smi
1 cpu版本的安装
直接pip install tensorflow即可
2 gpu版本的安装
1 pip install tensorflow-gpu
2 安装cuda-9.0 [一定是9.0版本,否则会有问题]
网址:https://developer.nvidia.com/cuda-downloads
选择红框下的选项,选择9.0版本,然后进入此页面
按照上面指导安装即可。
3 cudnn-9.4.1安装
网址:https://developer.nvidia.com/cudnn
这里一定要注意,需要注册一个账号,但是我注册后收不到邮件,也是醉了。于是使用同事的账号下载了。
这个就不说了,选择下载9.4.1的版本就行。下载后是一个绿色版的文件夹。因为到下载到服务器,所以我使用了抓包的方式,将内容直接写入到服务器的文件中。解压后,如图。
cp lib64/lib* /usr/local/cuda-9.0/lib64/
cp include/* /usr/local/cuda-9.0/include/
设置path路径:
vim /etc/profile
写入:export PATH=/usr/local/cuda-9.0/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
source /etc/profile
3 测试:
如果结果类似于上图,那么就安装成功啦!