参考声明:
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)
之后,终端运行: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://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并不保证使用您的机器和系统。