在 PyCharm 中使用 远程的 多GPU CUDA + Tensorflow

在运行配置中,加入环境变量

CUDA_DEVICE_ORDER=PCI_BUS_ID
CUDA_VISIBLE_DEVICES=1
LD_LIBRARY_PATH=/usr/local/cuda/lib64

前两行为程序指定GPU。这在多 GPU,以及显示用 GPU 和计算用 GPU 分离的情况中,非常重要。防止程序调用显示用GPU 或者调用了过多的GPU。

第三行是 cuda 的路径。远程部署的时候,使用.bashrc里的信息,所以需要这样配置。

发表评论

电子邮件地址不会被公开。 必填项已用*标注

This site uses Akismet to reduce spam. Learn how your comment data is processed.