前提安装好CUDA和cuDNN 如果没有安装好CUDA和cuDNN请转移至 https://blog.csdn.net/suyongcai1234/article/details/90750940
pytorch安装
进入pytorch官网:https://pytorch.org/get-started/locally/
选择对应的CUDA版本
按照提示的pip命令在windows终端中运行即可
测试
在python代码中
import torch
import torchvision
没有报错即为测试成功