其他分享
首页 > 其他分享> > RuntimeError: CUDA error: no kernel image is available for execution on the device

RuntimeError: CUDA error: no kernel image is available for execution on the device

作者:互联网

RuntimeError: CUDA error: no kernel image is available for execution on the device

这个问题是cuda版本和torch版本不对应,如下图,我的服务器cuda版本是11.2:

我复现的代码要求的torch版本是1.6.0,但是查找pytorch官网发现,torch1.6.0对应的cuda最高只到10.2,如图:

 

所以,只能不按照复现要求,提高torch的版本来迎合服务器cuda11.1的版本,找到torch1.8.0有对应cuda的11.1版本:(其实安装11.0版本也可以的)

选择

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

下载就好 

标签:available,kernel,no,RuntimeError,torch,cu111,cuda,版本
来源: https://blog.csdn.net/qq_49874815/article/details/123591766