1、安装cuda
- 命令行输入nvidia-smi,查看驱动信息
从官网下载对应版本的安装程序 CUDA Toolkit Archive | NVIDIA Developer
安装完成后验证:
验证:输入nvcc --version 进行检查
2、安装cudnn
下载Cudann: cuDNN Download | NVIDIA Developer(需要先注册账号)
将压缩包解压至CUDA安装路径下(D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1)
验证:
进入 d:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
运行bandwidthTest.exe
输出结果Result = PASS,安装成功
3、安装GPU版Pytorch
进入pytroch官网:https://pytorch.org/get-started/previous-versions/
找到上面的对应版本,直接copy对应命令下载
用wheel这个命令会明显要快