指定GPU
CUDA_VISIBLE_DEVICES=-1 python my.py
运行程序时不使用GPU
CUDA_VISIBLE_DEVICES=0,2,3
编号为0,2,3的GPU对程序可见,在代码中gpu[0]指的是第0块,gpu[1]指的是第2块,gpu[2]指的是第3块
CUDA_VISIBLE_DEVICES=2,0,3
编号为0,2,3的GPU对程序可见,在代码中gpu[0]指的是第2块,gpu[1]指的是第0块,gpu[2]指的是第3块
CUDA_VISIBLE_DEVICES=1 python your_file.py
CUDA_VISIBLE_DEVICES=-1 nohup python my.py &
查看GPU占用
nvidia-smi
网友评论