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