PyCharm远程执行TensorFlow代码
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/