1、卸载cuda8.0
cd /usr/local/cuda_8.0/bin
sudo ./uninstall_cuda_8.0.pl
2、安装cuda8.0时的显卡驱动不用卸载,nvidia-smi查看驱动版本。
3、下载并安装cuda9.0
链接:https://pan.baidu.com/s/1ju_wjXXy__RPH5o0yunDMw 密码:4rrb
sh cuda_9.0.176_384.81_linux-run
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n
nvcc -V检查安装,安装成功显示如下信息:
(base) [root@vm-21b-01xp8 ~]# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176
4、下载cudnn7.3
链接:https://pan.baidu.com/s/1vDUHMXxawqlHAX4Qvafh5w 密码:elpo
将文件夹中lib64和include的文件分别移动到/usr/local/cuda-9.0/lib64 和 include文件夹中。
sudo cp cudnn.h /usr/local/cuda-9.0/include
sudo cp ../lib64/* /usr/local/cuda-9.0/lib64
5、安装tensorflow-gpu确认是否配置成功。
6、有一些教程说需要以下操作:
本人并未进行如上操作,tensorflow-gpu完美运行。