caffe2安装步骤

caffe2安装步骤caffe 安装记录依赖项安装 sudoapt getupdatesud getinstall yno install recommends build essential cmake git libgoogle glog dev libprotobuf dev protobuf compiler python dev python pip

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

(0)
上一篇 2026年3月18日 上午9:03
下一篇 2026年3月18日 上午9:03


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号