在 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里的信息,所以需要这样配置。

发表回复

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

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据