指定运行代码的GPU
指定用第二个GPU运行代码:
import os os.environ['CUDA_VISIBLE_DEVICES']='1'
或者在命令行终端输入的命令前加上CUDA_VISIBLE_DEVICES= *
CUDA_VISIBLE_DEVICES=0 python your_file.py
指定用第二个GPU运行代码:
import os os.environ['CUDA_VISIBLE_DEVICES']='1'
或者在命令行终端输入的命令前加上CUDA_VISIBLE_DEVICES= *
CUDA_VISIBLE_DEVICES=0 python your_file.py
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 三行代码完成国际化适配,妙~啊~
· .NET Core 中如何实现缓存的预热?