问题来源
该问题源于项目工程自动下载的torch版本与cuda11.4支持的torch不兼容
那么问题就在于如何找到cuda11.4兼容的torch版本?在官网中寻找
一.选择与系统和cuda匹配的下载地址
首先打开pytorch官网
二.下载
按照图示找到对应cuda的torch下载指令,需要注意的是这个链接https://download.pytorch.org/whl/cu113/torch_stable.html是一个下载地址的展示,而不是一个特定的下载地址
当把Run this Command输入到命令行时,会弹出系统指定的whl的下载链接,当在线下载网速缓慢的时候,可以在本地下载好了再拷贝到服务器中
三.安装
pip install torch-1.10.1+cu113-cp37-cp37m-linux_x86_64.whl