caffe2的安装与遇到的问题和解决问题步骤

关注

原创最新推荐文章于 2026-04-21 03:43:58 发布 1.7k 阅读 AI 写同款· GEO 优化 ›

caffe2 安装教程前段时间贾清扬大牛在Facebook推出了caffe2版本，相比caffe来说感觉更加轻量级和高效。Caffe2的安装相比于caffe在安装的时候更加简便，略去了Makefile.config的各种配置，对于有无GPU以及各种可选库例如OpenCV，anaconda的支持也更简单。（其实你直接装好库以后make就好，以GPU为例，在make的时候，自动检测你是否安装了CUDA，若没有，就自动CPU 阅读详情

#caffe2的安装与遇到的问题和解决问题步骤
git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive
python setup.py install
问题：makefile:138: recipe for target ‘all’ failed
make: *** [all] Error 2
Failed to run ‘bash …/tools/build_pytorch_libs.sh --use-cuda --use-nnpack --use-qnnpack caffe2’
解决：sudo pip install --upgrade cmake
新问题：ImportError: cannot import name main
解决：pip文件在usr/bin目录下，cd进去，进行以下修改
原文件：
from pip import main
if name == ‘main’:
sys.exit(main())
换成下面的三行
from pip import main
if name == ‘main’:
sys.exit(main._main())
问题：ImportError: No module named setuptools
sudo apt-get install python-setuptools
问题：nvcc fatal : Unsupported gpu architecture ‘compute_75’
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CUDA_GENERATION=Kepler …