1.查看本机支持的CUDA版本
cmd:nvidia-smi
2.安装本机支持的CUDA版本
CUDA安装教程
GPU, CUDA,cuDNN三者的关系总结
3.使用anaconda创建虚拟环境
3.在anaconda对应的虚拟环境中安装CUDA对应的pytorch版本
anaconda安装pytorch(anaconda3,Windows10)
4.以上安装的pythorch为cpu版本
Pytorch最全安装教程
5.在对应的环境中安装torch和torchversion的GPU版本
报错:Torch not compiled with CUDA enabled看这一篇就足够了