import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # RTX2080Ti
######################################################
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2" # 2 *Tesla T4
指定GPU训练
最新推荐文章于 2024-07-11 11:51:31 发布
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # RTX2080Ti
######################################################
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2" # 2 *Tesla T4