关于pytorch的问题(RuntimeError: Could not run ‘aten::empty_strided‘ with arguments from the ‘CUDA‘ backe)
·
在运行已训练好的模型时,出现了一下错误:
RuntimeError: Could not run 'aten::empty_strided' with arguments from the 'CUDA' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'aten::empty_strided' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradNestedTensor, UNKNOWN_TENSOR_TYPE_ID, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, Autocast, Batched, VmapMode].
CPU: registered at aten\src\ATen\RegisterCPU.cpp:5925 [kernel]
BackendSelect: registered at aten\src\ATen\RegisterBackendSelect.cpp:596 [kernel]
Named: registered at ..\aten\src\ATen\core\NamedRegistrations.cpp:7 [backend fallback]
AutogradOther: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradCPU: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradCUDA: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradXLA: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradNestedTensor: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
UNKNOWN_TENSOR_TYPE_ID: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradPrivateUse1: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradPrivateUse2: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
AutogradPrivateUse3: registered at ..\torch\csrc\autograd\generated\VariableType_0.cpp:9273 [autograd kernel]
Tracer: registered at ..\torch\csrc\autograd\generated\TraceType_0.cpp:10499 [kernel]
Autocast: fallthrough registered at ..\aten\src\ATen\autocast_mode.cpp:250 [backend fallback]
Batched: registered at ..\aten\src\ATen\BatchingRegistrations.cpp:1016 [backend fallback]
VmapMode: fallthrough registered at ..\aten\src\ATen\VmapModeRegistrations.cpp:33 [backend fallback]
进程已结束,退出代码为 1
经过排查发现是安装的pytorch为cpu版本, 所以出现以上报错
import torch
print(torch.cuda.is_available())
print(torch.__version__)
结果为:
False
1.8.1+cpu
因此需要重新下载pytorch
建议参考这篇文章面向Windows的Pytorch完整安装教程_window pytorch 安装-CSDN博客
Pytorch1.8.1版本离线安装教程 作者:不当拉跨先生文章变vip了,可以看上面的
我使用的whl文件
下载网址:https://download.pytorch.org/whl/torch_stable.html
因为cuda为10.2
所以下载的是torch-1.8.1+cu102-cp37-cp37m-win_amd64
重新安装后再次运行
import torch
print(torch.cuda.is_available())
print(torch.__version__)
运行结果为
True
1.8.1 +cu102
则安装成功
再次运行代码,则能成功运行
更多推荐


所有评论(0)