PyTorch指定GPU进行训练 代码的路 2022-08-18 1,150 阅读1分钟 原文链接 1 直接在终端中设定: CUDA_VISIBLE_DEVICES=1 python main.py 2 python代码中设定: import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" 学习更多编程知识,请关注我的公众号: 代码的路