服务器指定GPU跑程序

最近在跑模型的过程中，发现服务器的前几个GPU总是被占用，故只能指定后面几个空闲GPU运行。

1. 在.py中指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "3"  　#指定第三块gpu

2. 在命令行指定

CUDA_VISIBLE_DEVICES = 2, 3, 4  bash your_file.sh

3. 在环境变量中指定

#打开.bashrc，加入下面代码
export  CUDA_VISIBLE_DEVICES = 2, 3, 4