在centos7下安装tensorflow-gpu 1.8版本教程

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 测试:

       

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