指定运行代码的GPU

指定用第二个GPU运行代码:

import os
os.environ['CUDA_VISIBLE_DEVICES']='1'

 

或者在命令行终端输入的命令前加上CUDA_VISIBLE_DEVICES= *

CUDA_VISIBLE_DEVICES=0    python  your_file.py

 

posted @   鱼儿冒个泡  阅读(101)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 三行代码完成国际化适配,妙~啊~
· .NET Core 中如何实现缓存的预热?
点击右上角即可分享
微信分享提示