TensorFlow-软件安装-CentOS下安装TensorFlow-附卸载-附TensorFlow网页-新手适用

参考声明:

https://blog.csdn.net/j790675692/article/details/78242231

https://blog.csdn.net/Oh_My_Fish/article/details/78861867

0.简介:

实验室的服务器之前安装过CUDA9.1,初始想法是,安装了相应版本的cuDNN。尝试之后,返现TensorFlow-GPU版本安装后无法使用,表现为import tensorflow之后报错。

多次尝试之后,确定无法使用,决定卸载CUDA9.1,重新安装。最终,基于CUDA9.0+ cuDNN 7.0.5环境,安装了TensorFlow-GPU版本。

1.驱动:

根据网上的教程,需要先到官网,根据显卡型号现在并安装NVIDIA的驱动。

由于实验室服务器之前安装过驱动,未保留过程截图,这里不再详细叙述。

参考教程:https://blog.csdn.net/j790675692/article/details/78242231

2.安装CUDA:

点击链接,下载CUDA9.0(注意检查一下版本是否为9.0)

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=runfilelocal

之后,终端运行:sudo sh cuda_9.0.176_384.81_linux.run

运行结束之后,会出现如下界面:

按Q退出后,就进入安装流程:

根据需要选择安装配置即可。碰到默认安装路径,若不更改,可直接敲回车跳过。

https://blog.csdn.net/Oh_My_Fish/article/details/78861867

这篇文章中提示:请务必只在“toolkit”时选择yes,否则将导致一系列显卡相关的问题,最坏的情况下,需要重新安装Nvidia显卡驱动。

3.修改环境变量:

执行 sudo vim /etc/profile,在export PATH 那行下面加上两行代码。

$ export PATH=/usr/local/cuda-8.0/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

编辑后,按esc退出,输入“:wq”退出。(注意冒号)

退出后,执行 source /etc/profile命令,使得环境变量生效。

4.安装cuDNN

cuDNN的安装,只需要将压缩包解压,并把文件覆盖到CUDA对应的目录中去即可:

$ tar -zxvf cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm
$ cd cuda
$ cp include/* /usr/local/cuda-9.0/inlcude/
$ cp lib64/lib* /usr/local/cuda-9.0/lib64/

5.安装TensorFlow:

执行命令:sudo pip3 install tensorflow-gpu

6.测试

使用python3运行下面程序:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)

7.CUDA卸载:

CUDA安装的方式不同,卸载的方式也不同。

  • 如果使用的是runfile文件安装的,默认会在/usr/local/cuda-9.0/bin/路径下,生成:uninstall_cuda_9.0.pl

卸载的时候,运行 sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl 即可。

  • 如果使用的是包管理方法安装的,那就不存在uninstall_cuda_9.0.pl,需要借助其他方式卸载:

 运行sudo yum remove <package_name>                      # Redhat/CentOS

方法来源于:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#handle-uninstallation

实际操作中,由于未知原因,上述两种方法都没有成功,借助下面的方法直接卸载:

输入命令:sudo yum erase cuda*

8.CUDA卸载相关参考:

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#handle-uninstallation

https://devtalk.nvidia.com/default/topic/992537/cuda-setup-and-installation/how-to-uninstall-cuda-8-0-on-rhel-6/

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#introduction

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-uninstallation

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

9.TensorFlow官网教程+CUDA9.0+cuDNN下载链接:

链接:https://pan.baidu.com/s/1obAF5g0g3ZueRlM6Guuzgw 密码:mu8s

注意:请读者检查系统版本,根据个人软硬件配置进行下载。本链接中的CUDA9.0+cuDNN并不保证使用您的机器和系统。

猜你喜欢

转载自blog.csdn.net/qq_17753903/article/details/82355759
今日推荐