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

 

总而言之就是安装的cuda, pytorch和当前的GPU型号并不匹配。

Solution

step 1 GPU

首先需要知道自己的GPU型号，

可以通过nvidia-smi获取。

step 2 CUDA version

通过去nvidia官网或者其他地方找到这款GPU对cuda版本的要求

 

pip install torchvision==0.13.1

pip 

pip install torchvision==0.13.1

 

   JUPYTER NOTEBOOK 运行成功

 but 出现了新的问题: