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

最新推荐文章于 2024-02-21 16:46:09 发布

Lotus_AI

最新推荐文章于 2024-02-21 16:46:09 发布

阅读量1.5k

点赞数 1

文章标签： caffe2 ubuntu

本文链接：https://blog.csdn.net/Julia_deeplearning/article/details/84104494

版权

本文详细记录了Caffe2的安装过程，包括使用PyTorch源代码仓库进行安装的具体步骤，以及在安装过程中可能遇到的问题及其解决方案。例如，解决makefile错误、ImportError异常、cmake版本升级、Python依赖包安装等常见问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#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 …