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

记录一下这个bug

0x01 bug由来

  疫情原因,大家都没有返校,而我回来了,正赶上服务器需要重装,这个重任落到了我的肩膀上

——如何走上运维之路

0x02 环境介绍

  • ubuntu 16.04(应该是)
  • 显卡型号Tesla k40m
  • python版本:3.6.10
  • pytorch版本:1.5.0

0x03 bug描述

  tensor可以放到gpu,但是不可以使用。从下图可以看到,我把a放到cuda上是没有问题的,但是在查看a的内容,也就是使用a的时候出现了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_第1张图片

0x04 解决方案

  在网上找了好久,后来在他github的issue里找到的答案:Version 1.3 no longer supporting Tesla K40m? #30532。所以我们的解决方案就是降低pytorch的版本,emmm……得降到1.2。所以,使用conda降低pytorch版本即可。

conda install pytorch=1.2

0x05 验证结果

  嗒哒~
RuntimeError: CUDA error: no kernel image is available for execution on the device_第2张图片

你可能感兴趣的:(Pytorch,Python,pytorch)