caffe安装记录
依赖项安装
sudo apt-get update sudo apt-get install -y --no-install-recommends \ build-essential \ cmake \ git \ libgoogle-glog-dev \ libprotobuf-dev \ protobuf-compiler \ python-dev \ python-pip sudo pip install numpy protobuf
源码下载及安装
git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2 make && cd build && sudo make install
注意:若在github上用zip格式下载,可能会出现若干第三方包(caffe2/third_party/)下载不下来的错误,因此请使用git clone,即上面的代码进行下载和安装;
测试安装是否成功:
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure" 输出success则安装成功
GPU测试:
python -m caffe2.python.operator_test.relu_op_test
成功:
0x7ff19b6237d0>], engine=
u'CUDNN') . ---------------------------------------------------------------------- Ran
1 test
in
4.858s OK
错误:Cudnn.h: No such file or directory
报错:
In file included from /home/hw/main-data/00_softwares/caffe2/caffe2/core/context_gpu.cu:10:0: /home/hw/main-data/00_softwares/caffe2/caffe2/core/common_cudnn.h:7:19: fatal error: cudnn.h: No such file or directory
cuDNN v7.0 Library for Linux #解压文件 tar -zxvf cudnn-6.5-linux-x64-v2.tgz #切换路径 cd cudnn-6.5-linux-x64-v2 #复制lib文件到cuda安装路径下的lib64/ sudo cp lib* /usr/local/cuda/lib64/ #复制头文件 sudo cp cudnn.h /usr/local/cuda/include/
建议:直接切换到su管理员权限,进行复制文件操作;网上说的需要更新软链接,我装的时候没有用上,补充在下边:
更新软连接
cd /usr/local/cuda/lib64/ sudo rm -rf libcudnn.so libcudnn.so.6.5 sudo ln -s libcudnn.so.6.5.48 libcudnn.so.6.5 sudo ln -s libcudnn.so.6.5 libcudnn.so
错误:No module named past.builtins
[global] index-url = http://mirrors.aliyun.com/pypi/simple/ [install] trusted-host=mirrors.aliyun.com
错误:No module named hypothesis
错误:在build文件夹下运行示例成功,其他文件夹错误
环境变量未设置好,python运行时不知道搜索路径,请做如下操作
echo $PYTHONPATH export PYTHONPATH=/usr/local:$PYTHONPATH export PYTHONPATH=$PYTHONPATH:/home/hw/main-data/00_softwares/caffe2/build echo $LD_LIBRARY_PATH export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
如果还是不行:
vim / etc/bash.bashrc 末尾添加 export PYTHONPATH=/usr/local:$PYTHONPATH export PYTHONPATH=$PYTHONPATH:/home/hw/main-data/00_softwares/caffe2/build export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/217664.html原文链接:https://javaforall.net
