新买的RTX2070到了,一顿操作后装的是cuda11.0,而pytorch目前支持的最高cuda版本为10.2;无奈需要重装,本想在不卸载cuda11.0的条件下安装10.2,一顿操作猛如虎,结果运行pytorch还是报错,无奈选择卸载11.0;
1.卸载cuda11.0
进入控制面板,卸载程序,将带有11.0版本号的NVIDIA CUDA逐个卸载(本人卸载时忘了截图了,这里截图以10.2为例)
打开计算机-属性-高级系统设置-环境变量-系统变量,将系统变量中存在CUDA11.0的路径都删除
2.安装cuda10.2
CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive
下载工具包,安装CUDA10.2
参考资料:PyTorch(GPU版)详细安装教程.
安装完成后 ,输入nvcc -V检查是否安装成功;
3.安装CUDNN
关于为什么要安装CUDNN,参考以下资料
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?.
CUDNN下载:https://developer.nvidia.com/rdp/cudnn-download,这里需要注册NVIDIA才可以下载,
下载完成后,解压文件,改名为CUDNN文件,剪贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2中
4.安装pytorch
下载命令行在官网可以看到:https://pytorch.org/
找到对应版本,命令行输入下载
下载完成后,torch.cuda.is_available()显示True即安装成功。
踩过好多坑,终于下载成功了~~