win10 安装Cuda11.0 + cudnn + pytorch
卸载CUDA(以前安装过cuda,必须要卸载干净)
1. 首先是软件卸载,按照名字排序,以下两个红框内的别卸载,其他全部卸载。有的教程说GeForce Experience也别卸载,我卸载了也没啥影响。
2. 使用杀毒软件请理注册表。
3. 删除三个文件夹【C:\ProgramData】,【C:\Program Files】,【C:\Program Files (x86)】中的“NVIDIA Corporation”和“NVIDIA GPU Computing Toolkit”。
4. 如果文件无法删除,打开服务管理器(win10搜索“计算机管理”),停止以下服务。
5. 再将3中文件清理干净,重启电脑。
安装Visual Studio 2019
不安装VS,有些功能无法正常使用。选择C++的桌面开发
安装CUDA
1. 先查看电脑支持的版本,打开NIVDIA控制面板,菜单栏->帮助->系统信息->组件
看到是11.0版本,好像是11.0及以下都可以的,但我还是选择装11.0。
2. 打开官网(https://developer.nvidia.com/cuda-toolkit-archive),选择11.0
网页下载很慢,使用迅雷下载local版本
3. 运行安装包,直接安装精简版本
4. Path环境变量添加
5. 测试,打开cmd
nvcc -V
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite
bandwidthTest.exe
deviceQuery.exe
安装cudnn
1. 官网下载(https://developer.nvidia.com/cudnn),登录账号,做问卷
2. 将解压文件里面的bin、include、lib文件夹复制到CUDA安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
安装pytorch
pytorch官网(https://pytorch.org/get-started/locally/)可以获得安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
下载可能比较慢,解决方法
换源 或 下载离线安装包(参考https://www.cnblogs.com/farewell-farewell/p/11096438.html)