指定使用第一块GPU,即编号为0
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
CUDA_VISIBLE_DEVICES=1 Only device 1 will be seen
CUDA_VISIBLE_DEVICES=0,1 Devices 0 and 1 will be visible
CUDA_VISIBLE_DEVICES=“0,1” Same as above, quotation marks are optional
CUDA_VISIBLE_DEVICES=0,2,3 Devices 0, 2, 3 will be visible
CUDA_VISIBLE_DEVICES=“” No GPU will be visible
据说也可以在终端执行程序时指定GPU,但未验证是否可以,指令如下:
$ CUDA_VISIBLE_DEVICES=1 python run_file.py