pytorch版本和gpu不匹配的问题记录

时间:2024-03-24 16:15:28

1.出现的问题
a. RuntimeError :cuDNN error CUDNN_STATUS_BAD_PARAM

b.RuntimeError: cuda error: no kernel image is available for execution on the device

c.一个警告,我两个GPU大小不一致,一块4G,一块12G.【看p2】
这个的解决方案是指定一块gpu运行。
关于a,b中的问题,我是重新安装了低版本的pytorch(torch1.3.0,torchvision0.4.1)
原先安装的是torch==1.5.1

pytorch版本和gpu不匹配的问题记录
pytorch版本和gpu不匹配的问题记录
pytorch版本和gpu不匹配的问题记录

2.电脑环境:

  • cuda: 10.1
  • gpu: 0:Quardo K2200 (4G); 1:Tesla K40c (12G)
  • torch == 1.5.1

3.其他问题
当torch==1.5.1的时候,我可以用Quardo K2200 (4G)这块GPU跑程序。但不能用 Tesla K40c (12G)这块。降低torch版本之后,可以用 Tesla K40c (12G)这块跑,但是不能用Quardo K2200 (4G)这块。囧 orz…