import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 改GPU编号使用0号显卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" # 改GPU编号使用0\1号两块显卡
指定GPU进行训练
最新推荐文章于 2024-07-19 16:36:18 发布
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 改GPU编号使用0号显卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" # 改GPU编号使用0\1号两块显卡