nvidia-smi
共有0,1,2三块gpu,processes显示GPU占用情况。
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “2”
选择指定的gpu = 2运行程序
CUDA_VISIBLE_DEVICES=0,1
在终端运行时指定gpu
nvidia-smi
共有0,1,2三块gpu,processes显示GPU占用情况。
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “2”
选择指定的gpu = 2运行程序
CUDA_VISIBLE_DEVICES=0,1
在终端运行时指定gpu