问题描述
在A环境可以运行的代码,到B环境就报错了
RuntimeError: CUDA error: no kernel image is available for execution on the device
解决过程
问了GPU有几种可能,
我的解决办法
升级了pytorch 和cudatoolkit版本即可(论文说是用Pytorch 1.7 cudatoolkit 10.2 但是我的机器上跑不通),最后通过 以下命令升级了版本问题就解决了
$ conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge