主要介绍了浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

服务器有多张显卡，一般是组里共用，分配好显卡和任务就体现公德了。除了在代码中指定使用的 GPU 编号，还可以直接设置可见 GPU 编号，使程序/用户只对部分 GPU 可见。 操作很简单，使用环境变量 CUDA_VISIBLE_DEVICES 即可。 具体来说，如果使用单卡运行 Python 脚本，则可输入 CUDA_VISIBLE_DEVICES=1 python my_scr ipt.py 脚本将只使用 GPU1。 在 .py 脚本和 Notebook 中设置，则 import os o