Tensorflow训练模型默认占满所有GPU---解决方案
在使用gpu服务器训练tensorflow模型时,总是占满显存!TensorFlow默认的是占用所有GPU因此我们需要手动设置使用的GPU编号以及单个GPU显存占用比例1.第一步需要在代码中开头加入import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 将设备中所有GPU按照os.environ["CUDA_VISIBLE_DEVICES"]=‘0’ # 使用0号gpu(想使用其他编号GPU,对应修改引号中的内容即可)os.env
原创
2020-08-06 10:01:33 ·
3208 阅读 ·
0 评论