显卡和cuda都要重新安装一次才可以
显卡下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn
cuda、显卡版本和tensorflow-gpu对应关系:https://blog.csdn.net/weixin_45284801/article/details/105920249
如何确认自己的GPU是否支持CUDA
登陆网址,https://developer.nvidia.com/cuda-gpus
要根据显卡确定要安装哪个cuda
cuda 9.0地址:
CUDA各版本官方下载地址
地址:https://developer.nvidia.com/cuda-toolkit-archive
根据cuda 版本 确定 加速库 cudnn 的版本 ,cudnn不是必须装的,但是装了可以提升1.5倍速度
cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。
卸载cuda和显卡驱动
cuda: /cuda/bin/uninstall_cuda_9.0.pl
显卡:/usr/bin/nvidia-uninstall
配置CUDA环境变量
sudo vim ~/.bashrc
在末尾加入:
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc
参考链接:
https://blog.csdn.net/qq_25387073/article/details/99623872