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

下载就好