2020
10-08
10-08
浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置
服务器有多张显卡,一般是组里共用,分配好显卡和任务就体现公德了。除了在代码中指定使用的GPU编号,还可以直接设置可见GPU编号,使程序/用户只对部分GPU可见。操作很简单,使用环境变量CUDA_VISIBLE_DEVICES即可。具体来说,如果使用单卡运行Python脚本,则可输入CUDA_VISIBLE_DEVICES=1pythonmy_script.py脚本将只使用GPU1。在.py脚本和Notebook中设置,则importosos.environ["CUDA_VISIBLE_DEVICES"]="2"...
继续阅读 >