首页 » 未分类 » PyCharm远程执行TensorFlow代码

PyCharm远程执行TensorFlow代码

作者: vForce 分类: 未分类 发布时间: 2017-06-27 18:32 ė浏览 23,931 次 6没有评论

PyCharm使用远程解释器执行的时候,由于使用的SSH链接是notty的,因此不会加载bashrc内的环境变量定义,所以如果直接跑TensorFlow代码会在加载CUDA时遇到如下报错:

ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory

要解决这个问题,就需要在Run/Debug Configuration里对应的Script下添加Envrionment Variables:

Name: LD_LIBRARY_PATH

Value: /usr/local/cuda-8.0/lib64/(此处应为对应的CUDA目录)

添加完毕后即可正常运行

本文出自 Tech Trace,转载时请注明出处及相应链接。

本文永久链接: https://www.qiujiahui.com/2017/06/27/pycharm%e8%bf%9c%e7%a8%8b%e6%89%a7%e8%a1%8ctensorflow%e4%bb%a3%e7%a0%81/

0

发表回复

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

Ɣ回顶部