原文链接
1 直接在终端中设定:
CUDA_VISIBLE_DEVICES=1 python main.py
2 python代码中设定:
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
学习更多编程知识,请关注我的公众号:
代码的路