2021年8月21日21:14:42
目前,pytorch 的稳定发布版还不支持cuda11.3
如要使用了30系列的显卡,安装了比较新的cuda,可以去下面的链接下载dev版本的pytorch
https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html
根据自己的python版本和os下载对应的版本。
可以ctrl + F快速定位。
这里我直接搜索113,就是对应cuda11.3 的版本。
下载后,使用pip install 命令即可安装成功。