本帖旨在记录自己平时遇到的各种关于os.environ的设置,不定期更新
0.使用方法:设置环境变量,自然是要放在代码最前面的(但是要先导入os包)。例如下面:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
1."CUDA_VISIBLE_DEVICES"
:自定义要使用的GPU(tensorflow中默认是全部使用的)
注意:GPU一般是从0开始编号的,可以通过nvidia-smi
查看,如下图:
本帖旨在记录自己平时遇到的各种关于os.environ的设置,不定期更新
0.使用方法:设置环境变量,自然是要放在代码最前面的(但是要先导入os包)。例如下面:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
1."CUDA_VISIBLE_DEVICES"
:自定义要使用的GPU(tensorflow中默认是全部使用的)
注意:GPU一般是从0开始编号的,可以通过nvidia-smi
查看,如下图: