cublas运行时错误:GPU程序无法在/pytorch/aten/src/THC/ www.example.com上执行THCBlas.cu:450

iugsix8n  于 2023-04-12  发布在  其他
关注(0)|答案(3)|浏览(296)

当我在colab GPU中运行Python脚本时,我得到了这个错误:
RuntimeError:cublas运行时错误:GPU程序无法在/pytorch/aten/src/THC/ www.example.com上执行THCBlas.cu:450
我该如何解决这个问题?
Pytorch版本:1.1.0,Cuda版本:9.0.
请让我知道如果任何其他细节是必需的。

n9vozmp4

n9vozmp41#

你需要升级你的torch。我也遇到了这个问题,升级torch后,这个问题就解决了。
我使用了这个命令:

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/

希望能成功!

blpfk2vs

blpfk2vs2#

您可以尝试使用CUDA版本10而不是9:
对于RTX 2080,您需要使用CUDA10,而不是CUDA 9.2

to94eoyn

to94eoyn3#

我更新了{torch}到1.13.1和cudatoolkit=10.1.243,问题解决了。我使用RTX-3070

相关问题