cuda 版本 :11.6
cudnn 版本:11.5
之前安装 torch 的时候由于网络原因找不到官方的 whl,导致 Pip 自动安装 cpu 版本的 torch
手动去官网下载下载链接中的 whl ,然后手动安装 whl
torch-1.12.0+cu116-cp37-cp37m-win_amd64
torchaudio-0.12.0+cu116-cp37-cp37m-win_amd64
torchvision-0.13.0+cu116-cp37-cp37m-win_amd64.whl