1、卸载原有的torch版本
可以在cmd输入如下命令
pip uninstall torch
或者也可以直接去环境下的Lib\site-packages文件下手动删除所有包含torch的文件(包括torchversion)
2、Pytorch官网下载gpu版本的torch
Pytorch官网地址:Pytorch官网
进官网后按如下图所示配置
然后进入红线标记的网站,下载对应的gpu版本
3、查找自己对应的版本
在cmd输入如下命令
pip debug --verbose
下载对应的torch和torchversion的whl文件
4、进入你安装whl文件的目录下进行pip安装
同时安装torchv-xxx.whl,完成安装
5、检测是否安装成功