1. 选择要用的GPU: CUDA_VISIBLE_DEVICES
服务器中有多个GPU,选择特定的GPU运行程序可在程序运行命令前使用:CUDA_VISIBLE_DEVICES=0
命令。0为服务器中的GPU编号,可以为0, 1, 2, 3等,表明对程序可见的GPU编号。
2.os.environ
设置系统环境变量 :os.environ[‘环境变量名称’]=‘环境变量值’
获取系统环境变量:os.environ[‘环境变量名称’]
其中 key 和 value 均为 string 类型
查看所有环境变量的key:os.environ.keys()
3. seed
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
seed的具体作用详见