指定GPU运行程序

2019-10-23 本文已影响0人习惯了千姿百态

方法1 在python程序中
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
方法2
运行时
CUDA_VISIBLE_DEVICES=1 python **.py
方法3
直接在环境变量中 export