调错 RuntimeError: CUDA error: the launch timed out and was terminated

主要是对于双显卡的电脑运行深度学习分割代码报错（运行 predict.py 时候报错）参考这一篇cuda运行错误：the launch timed out and was terminated in_Liber-coder的博客-CSDN博客该方法并没有解决代码报错考虑到是GPU报错，调用的问题我们更换调用的GPU换成独立显卡CUDA_VISIBLE_DEVICES='0'#表示调用0块GPU或者

搬砖班班长喜欢阿花

9350人浏览 · 2021-12-17 22:33:10

搬砖班班长喜欢阿花 · 2021-12-17 22:33:10 发布

主要是对于双显卡的电脑

运行深度学习分割代码报错（运行 predict.py 时候报错）

参考这一篇cuda运行错误：the launch timed out and was terminated in_Liber-coder的博客-CSDN博客

该方法并没有解决代码报错

考虑到是GPU报错，调用的问题

我们更换调用的GPU

换成独立显卡

CUDA_VISIBLE_DEVICES='0' #表示调用0块GPU

或者

import os

os.environ["CUDA_DEVICES_ORDER"]="PCI_BUS_IS"

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

没有报错，成功运行

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

LLM支持的AI Agent对话状态跟踪

随着人工智能技术的飞速发展，基于大语言模型（LLM）的AI Agent在对话系统中的应用越来越广泛。对话状态跟踪作为对话系统中的关键环节，其目的在于让AI Agent能够准确记录和理解对话的上下文信息，从而根据当前对话状态做出合理的回应。本文的范围涵盖了LLM支持的AI Agent对话状态跟踪的核心概念、算法原理、数学模型、项目实战、实际应用场景以及未来发展趋势等方面，旨在为读者提供一个全面而深入

2048 AI社区

提示工程架构师必备：领域驱动设计（DDD）落地实战指南，从理论到代码全流程！

在当今的软件开发领域，随着业务的不断发展和变化，软件系统变得越来越复杂。传统的开发方式往往注重数据和技术，而忽略了业务领域的本质。这就导致了软件系统与业务需求之间的脱节，代码难以理解和维护，开发效率低下。领域驱动设计（Domain - Driven Design，简称 DDD）正是为了解决这些问题而诞生的。它强调以业务领域为核心，通过深入理解业务领域的知识和规则，将业务概念和逻辑映射到软件系统中，