Ubuntu16.04 安装 pycaffe

最新推荐文章于 2023-03-03 18:14:50 发布

原创最新推荐文章于 2023-03-03 18:14:50 发布 · 682 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#Caffe #Python #Linux #机器学习

机器学习专栏收录该内容

3 篇文章

订阅专栏

本文档详细介绍了如何在Ubuntu16.04操作系统中安装PyCaffe，参照了OpenPose的安装指南，适用于配备GTX1050显卡、CUDA10.0和cuDNN7.5的环境。首先，需要确保安装了所有必要的依赖库和软件。通过特定命令检查Python模块版本，然后在Caffe源码根目录执行编译和安装步骤，最后设置环境变量以完成PyCaffe的配置。

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

Caffe 的安装参考 OpenPose Installation：Ubuntu16.04 (Desktop) + GTX1050 + CUDA10.0 + cuDNN7.5。

在 /caffe/python/ 下有一个 requirements.txt，根据里面的内容，所需的依赖库和软件。

sudo python -m pip install jupyter #用jupyter代替ipython,使用ipython也可以
sudo python -m pip install opencv-python
sudo python -m pip install scipy
sudo python -m pip install cython
sudo apt-get install python-imaging #PIL.image
sudo apt-get install python-matplotlib
sudo python -m pip install 'networkx==2.2' #python2.7需要执行
sudo python -m pip install scikit-image
sudo python -m pip install h5py
sudo python -m pip install leveldb
sudo python -m pip install nose
sudo python -m pip install pandas
sudo python -m pip install protobuf #通常已经有了
sudo python -m pip install python-gflags
sudo python -m pip install pyyaml

（ pip list | grep -i <name> 命令可以查看某个 python 模块的版本）

在 caffe 根目录下 make clean && make && make pycaffe

成功后添加环境变量

echo 'export PYTHONPATH=/path to your/caffe/python:$PYTHONPATH' >> ~/.bashrc
source ~/.bashrc