1. 添加搜索路徑
import sys sys.path.append('/home/dddzz/caffe/python')
2. 修改環境變量文件 .bashrc
export PYTHONPATH=/home/dddzz/caffe/python:$PYTHONPATH
3. 添加caffe根目錄
import sys caffe_root = "/home/dddzz/worksapce/caffe-ssd" os.chdir(caffe_root) sys.path.insert(0,os.path.join(caffe_root,'python'))
4. 添加 cuda 位置
run --> edit configurations
找到 enviroment variables, 添加:
name: LD_LIBRARY_PATH value: usr/local/cuda-9.0/lib64