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

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

你可能感兴趣的:(caffe2的安装与遇到的问题和解决问题步骤)