
在成功编译caffe的源码之后,可以在Python环境中使用caffe。
在Ubuntu环境下,打开python解释程序,输入import caffe时:出现以下错误
>>>import caffe
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: No module named caffe
二、解决思路
基本思路是把caffe中的python导入到解释器中
三、解决方法
第一种方法:设置环境变量
在终中输入:
export PYTHONPATH=~/caffe/python #caffe的路径下面的python
则该终端起作用,关掉终端后或重新打开一终端,则失效。
放到配置文件中,可以永久有效果,命令 *** 作如下:
A.把环境变量路径放到 ~/.bashrc文件中
sudo echo export PYTHONPATH="~/caffe/python" >>~/.bashrc
B.使环境变量生效
source ~/.bashrc
第二种方法:通过代码来实现
在每个python代码中使用以下代码: (这个方法在写python代码时有用)
caffe_root = '~/caffe/python '
import sys
sys.path.insert(0, caffe_root + 'python')
import caffe
步骤1:安装所需的依赖库
1
命令如下。
(1)sudo apt-get update
(2)sudo apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
libgoogle-glog-dev \
libprotobuf-dev \
protobuf-compiler \
python-dev \
python-pip
(3)sudo pip install numpy protobuf
END
步骤2:可选的GPU支持
1
概述。如果您打算使用GPU而不是CPU,那么您应该安装NVIDIA CUDA 8和cuDNN v5.1或v6.0,这是GPU加速的深度神经网络的原生类型库。
2
首先更新您的显卡驱动程序(NVIDIA CUDA 8)! 否则您可能会遇到各种难以诊断的错误。
(1)sudo apt-get update &&sudo apt-get install wget -y --no-install-recommends
(2)wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
(3)sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
(4)sudo apt-get update
(5)sudo apt-get install cuda
2、适用Ubuntu 14.04的命令行。
(1)sudo apt-get update &&sudo apt-get install wget -y --no-install-recommends
(2)wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb"
(3)sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb
(4)sudo apt-get update
(5)sudo apt-get install cuda
3
安装cuDNN(适用所有Ubuntu版本)。
(1)CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
(2)wget ${CUDNN_URL}
(3)sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
(4)rm cudnn-8.0-linux-x64-v5.1.tgz &&sudo ldconfig
END
步骤3:可选依赖库
libgflags库安装。注意:libgflags2是用于Ubuntu 14.04;libgflags-dev是用于Ubuntu 16.04。这两个不能搞混,否则安装会报错。
(1)适用Ubuntu 14.04的命令行。
sudo apt-get install -y --no-install-recommends libgflags2
(2)适用Ubuntu 16.04的命令行。
sudo apt-get install -y --no-install-recommends libgflags-dev
安装其他可选依赖库(适合Ubuntu 14.04和16.04)。
(1)sudo apt-get install -y --no-install-recommends \
libgtest-dev \
libiomp-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libopenmpi-dev \
libsnappy-dev \
openmpi-bin \
openmpi-doc \
python-pydot
(2)sudo pip install \
flask \
future \
graphviz \
hypothesis \
jupyter \
matplotlib \
pydot python-nvd3 \
pyyaml \
requests \
scikit-image \
scipy \
setuptools \
six \
tornado
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)