Linux指定GPU跑程序

Linux指定GPU跑程序

一、单卡运行

1. 在train.py主程序指定

import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = gpu_ids
os.environ[“CUDA_VISIBLE_DEVICES”] = “2”
os.environ必须在import torch之前，不然程序运行的时候默认使用gpu0来训练程序

2. linux终端命令行执行

CUDA_VISIBLE_DEVICES=gpu_ids python3 train.py

（1）CUDA_VISIBLE_DEVICES=1,2 // 仅使用device 1和 device2
（2）CUDA_VISIBLE_DEVICES=0,2,3 // 仅使用device 0, device2和device3
（3）CUDA_VISIBLE_DEVICES=2,0,3 // 仅使用device0, device2和device3
（4）CUDA_VISIBLE_DEVICES后面参数依次是设置gpu[0]，gpu[1], gpu[2]…等的device编号。

0,2,3意思是gpu[0]指向device0, gpu[1], 指向devcie2, gpu[2]指向device3；
而2,0,3意思是gpu[0]指向device2, gpu[1], 指向devcie0, gpu[2]指向device3；

（5）如果当前主机有5张显卡，默认情况下5个device对程序都