Torch GPU版安装

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这个命令会明显要快