weixin_43147947的博客

原创 os.environ

1,“CUDA_VISIBLE_DEVICES”：自定义要使用的GPU（tensorflow中默认是全部使用的） 注意：GPU一般是从0开始编号的，可以通过nvidia-smi查看 import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" #仅使用GPU2（也就是第3块GPU） os.environ["CUDA_VISIBLE_DEVICES"] = ...