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

               40、Tensorflow:在centos7下安装tensorflow-gpu 1.8版本教程_第1张图片

选择红框下的选项,选择9.0版本,然后进入此页面

        40、Tensorflow:在centos7下安装tensorflow-gpu 1.8版本教程_第2张图片

 

按照上面指导安装即可。

3 cudnn-9.4.1安装

网址:https://developer.nvidia.com/cudnn

40、Tensorflow:在centos7下安装tensorflow-gpu 1.8版本教程_第3张图片

这里一定要注意,需要注册一个账号,但是我注册后收不到邮件,也是醉了。于是使用同事的账号下载了。

这个就不说了,选择下载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 测试:

       40、Tensorflow:在centos7下安装tensorflow-gpu 1.8版本教程_第4张图片

如果结果类似于上图,那么就安装成功啦!

你可能感兴趣的:(Tensorflow)