2017年12月16日
TensorFlow设置GPU占用量
默认开启Tensorflow的session之后，就会占用几乎所有的显存，进行如下设置即可： 指定GPU编号： import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 或者在脚本或者命令行中指定 export CUDA_VISIBLE_DEVICES=1 Read More
posted @ 2017-12-16 20:46 rainsoul Views(4091) Comments(0) Diggs(0) Edit