指定gpu两种方式:
1、设置环境变量,直接在命令行中使用,
set CUDA_VISIBLE_DEVICES=1
终端执行脚本,
CUDA_VISIBLE_DEVICES=1 python script.py
2、在程序中指定,os
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "5"
Ref:
1、https://www.jianshu.com/p/9772de4ceb0e
2、https://www.cnblogs.com/darkknightzh/p/6591923.html#_label0