TensorFlow Memo

TensorFlow Memo

单机多 GPU 使用单个 GPU 的一种方式

1
CUDA_VISIBLE_DEVICES=0 python my_script.py

这种方式, TF 的计算和显存再用都只是使用 0 号 GPU, 不需要对代码做出改动. 推荐使用.