安装CUDA的步骤参考博客链接:https://blog.csdn.net/hizengbiao/article/details/88625044
1、CUDA版本号一定需要和nvidia对应,否则CUDA安装过程会报错。
2. tensorflow-gpu、python、cudnn、cuda之间的依赖关系非常强,所以版本一定要对应。
3. nvidia显卡驱动卸载安装方法。
卸载显卡命令: yum remove nvidia-*
下载显卡驱动(当前版本为410.78改动需要的版本号): wget http://cn.download.nvidia.com/XFree86/Linux-x86_64/410.78/NVIDIA-Linux-x86_64-410.78.run
安装显卡驱动:sh ./NVIDIA-Linux-x86_64-410.78.run
验证是否安装成功:nvidia-smi