一行代码解决PyTorch训练模型时突然出现的For debugging consider passing CUDA_LAUNCH_BLOCKING=1报错
一、问题描述
今天在调试模型的代码,然后代码一直运行得好好地,就突然出现了一下的错误:
RuntimeError: CUDA error: invalid device ordinal
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
觉得十分诡异,前面运行的时候没出现这个Error,但是后面点击运行的时候就出现了,而且多次点击运行,都会报的这个错。
上面的错误大概就是说我的CUDA设备序号出错。例如,训练模型的机器上只有4张显卡,那么CUDA设备序号分别是0、1、2、3。当你在程序中使用4、5或者更加大的数字时,这个错误信息就会自动地报出来。
令我觉得诡异的地方是,在我运行模型的机器中,明明有4个可以使用的CUDA设备,也就是拥有4张显卡,而我不能使用1、2、3对应的CUDA设备,然后就报了上面的这个错误。所以我觉得十分奇怪。
二、解决方法
首先,我观察了我设置程序使用CUDA设备的代码:
-
os.environ['CUDA_VISIBLE_DEVICES'] = '%d' % m_gpu
-
torch.cuda.set_device(m_gpu)
-
torch.cuda.is_available()
-
torch.cuda.current_device()
其中上边的m_gpu是设置CUDA设备序号的变量。
然后我发现在设置CUDA设备序号时,如果调用了下面的代码,就不会出现上门的报错信息:
torch.cuda.device_count()
最终,设置CUDA设备序号的代码改为:
-
torch.cuda.device_count()
-
os.environ['CUDA_VISIBLE_DEVICES'] = '%d' % m_gpu
-
torch.cuda.set_device(m_gpu)
-
torch.cuda.is_available()
-
torch.cuda.current_device()
这样就解决了我上面所遇到的问题。
三、设备信息
NVIDIA驱动:NVIDIA-SMI 515.65.01 Driver Version: 515.65.01 CUDA Version: 11.7
Python版本:Python 3.6.9
PyTorch版本:1.10.2
Linux系统:Ubuntu 18.04.6 LTS (GNU/Linux 4.15.0-041500-generic x86_64)
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhghbbka
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01 -
怎样阻止微信小程序自动打开
PHP中文网 06-13