python os.environ 环境变量

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

# 目前所有环境变量
os.environ.keys()

# 设置环境变量
os.environ['JAVA_HOME'] = "/home/user/jdk1.8.0_212"
os.environ["JAVA_HOME"] = "/usr/local/java"

# 选择gpu
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
# 禁用gpu
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'