在终端可以import caffe但是在pycharm中No module named caffe

来源:互联网 发布:dnf如果合约到期知乎 编辑:程序博客网 时间:2024/06/06 00:09

一.在python脚本中import caffe 添加

sys.path.append('/home/wenyangming/caffe_new/caffe/python')

但是仍然会遇到 cannot libcudart 的问题,这时候可以用如下解决方法:
  1. 进入pycharm-community-2016.2/bin
  2. 打开pycharm.sh
  3. 185行左右
# 找到这行LD_LIBRARY_PATH="$IDE_BIN_HOME:$LD_LIBRARY_PATH" "$JAVA_BIN" \
  • 1
  • 2
  • 1
  • 2
# 添加环境变量LD_LIBRARY_PATH='/usr/local/cuda/lib64'
  • 1
  • 2
  • 1
  • 2
# 修改后LD_LIBRARY_PATH='/usr/local/cuda/lib64'LD_LIBRARY_PATH="$IDE_BIN_HOME:$LD_LIBRARY_PATH" "$JAVA_BIN" \
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3

4.重新启动pycharm,之前报错的脚本就能正常运行了


二.另一方案直接在terminal中,进行启动,就会自动继承来自系统后台的环境变量

1 0
原创粉丝点击