1.先更新nvidia驱动 NVIDIA 引领人工智能计算 | NVIDIA
2.确定要用的cuda型号 下载并安装 我用的10.2 CUDA Toolkit Archive | NVIDIA Developer
3.去 pytorch官网 获取对应cuda型号的pytorch版本的conda或pip口令 Start Locally | PyTorch
我用的conda 在conda环境下新建一个pytorch环境 然后运行conda口令
4.若要更改torchvision版本 在清华源 Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 下载对应的版本至本地 然后用 conda install --offline 对应的安装包文件名字来安装 如我的是conda install --offline F:torch torchvisiontorchvision-0.9.1-py36_cu102.tar.bz2
5.测试 pip list看版本 torch.cuda.is_available()检测cuda是否可用
