cuda+cudnn+pytorch安装与卸载

1.cuda下载地址:https://developer.nvidia.com/zh-cn/cuda-downloads?target_os=Linux

  cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-archive

2.cuda卸载,注意版本号要改对,第一条执行完命令行会提示卸载成功,第二条是删除残余文件

sudo /usr/local/cuda-10.1/bin/cuda-uninstaller 

sudo rm -rf /usr/local/cuda-10.1/

3.cudnn卸载

  1. sudo rm -rf /usr/local/cuda/include/cudnn.h

  2. sudo rm -rf /usr/local/cuda/lib64/libcudnn

卸载完后检查是否卸载成功,没输出就说明成功了:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

3.cuda安装:https://blog.csdn.net/QLULIBIN/article/details/78714596 

安装命令:./cuda_10.0.130_410.48_linux.run

4.cudnn安装,只看cudnn部分:https://blog.csdn.net/wanzhen4330/article/details/81699769

 

7.pytorch安装:https://pytorch.org/get-started/previous-versions/  注意命令去掉-c pytorch才会从清华源下

pytorch whl 文件下载地址:https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(AI,pytorch,神经网络,cuda)