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+torchvision\torchvision-0.9.1-py36_cu102.tar.bz2）
5.测试：pip list看版本 ；torch.cuda.is_available()检测cuda是否可用
版权声明：本文为weirdo_coder原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。