通过代码指定：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 1, 2"
注意最好放在代码首行，显卡从第0块开始
通过终端命令指定：
CUDA_VISIBLE_DEVICES=1,2,3 python train.py
文章介绍了如何通过在代码中设置环境变量`CUDA_VISIBLE_DEVICES`或在终端中使用命令来指定GPU的使用，特别是指定了从第0块开始的显卡编号。这种方法对于多GPU环境的管理和任务分配至关重要。
扫一扫
808
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
