参考这个博主的教程即可安装成功
https://blog.csdn.net/WhiffeYF/article/details/113527759?spm=1001.2014.3001.5502
主要说一下过程中遇到的一些问题
1.AssertionError Torch not compiled with CUDA enabled
出现这个问题先查看一下自己的torch是不是cuda版本的,因为换源之后,国内下载源可能下载的都是cpu版本的
查看torch.cuda.is_available()返回false的话就要卸载安装的torch,重新安装
可以在https://download.pytorch.org/whl/torch_stable.html下载适合自己的版本,然后使用pip安装