TensorFlow Memo 发表于 2016-01-01 分类于 Memo TensorFlow Memo 单机多 GPU 使用单个 GPU 的一种方式 1CUDA_VISIBLE_DEVICES=0 python my_script.py 这种方式,TF 的计算和显存再用都只是使用 0 号 GPU, 不需要对代码做出改动。推荐使用。