赞
踩
目录
1.源代码中添加
2.在gpu中运行时添加
3. 不添加源代码
import os# 指定使用0,1,2三块卡os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"
CUDA_VISIBLE_DEVICES=0,1,2