当我在colab GPU中运行Python脚本时,我得到了这个错误:RuntimeError:cublas运行时错误:GPU程序无法在/pytorch/aten/src/THC/ www.example.com上执行THCBlas.cu:450我该如何解决这个问题?Pytorch版本:1.1.0,Cuda版本:9.0.请让我知道如果任何其他细节是必需的。
n9vozmp41#
你需要升级你的torch。我也遇到了这个问题,升级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/
希望能成功!
blpfk2vs2#
您可以尝试使用CUDA版本10而不是9:对于RTX 2080,您需要使用CUDA10,而不是CUDA 9.2
to94eoyn3#
我更新了{torch}到1.13.1和cudatoolkit=10.1.243,问题解决了。我使用RTX-3070
3条答案
按热度按时间n9vozmp41#
你需要升级你的
torch
。我也遇到了这个问题,升级torch
后,这个问题就解决了。我使用了这个命令:
希望能成功!
blpfk2vs2#
您可以尝试使用CUDA版本10而不是9:
对于RTX 2080,您需要使用CUDA10,而不是CUDA 9.2
to94eoyn3#
我更新了{torch}到1.13.1和cudatoolkit=10.1.243,问题解决了。我使用RTX-3070