import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “0, 1”
程序中指定GPU代码
最新推荐文章于 2024-06-28 22:45:22 发布
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “0, 1”