ubuntu18.04安装caffe
时间: 2023-04-29 10:03:21 浏览: 238
安装Caffe在Ubuntu 18.04上的步骤如下:
1. 安装依赖项:sudo apt-get update && sudo apt-get install -y build-essential cmake git pkg-config libatlas-base-dev libboost-all-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libprotobuf-dev libsnappy-dev protobuf-compiler python-dev python-pip python-numpy python-scipy python-matplotlib python-sklearn python-h5py python-protobuf python-skimage python-opencv python-pandas python-yaml
2. 下载Caffe源码:git clone https://github.com/BVLC/caffe.git
3. 进入Caffe目录:cd caffe
4. 创建Makefile.config文件:cp Makefile.config.example Makefile.config
5. 修改Makefile.config,如果需要使用CPU支持,将CPU_ONLY设置为1,并且注释掉相应的GPU支持部分
6. 编译安装:make all -j8 && make install
7. 安装python接口:make pycaffe
8. 添加环境变量:echo "export PYTHONPATH=$(pwd)/python:$PYTHONPATH" >> ~/.bashrc && source ~/.bashrc
9. 验证安装:python -c "import caffe; print(caffe.__version__)"
请注意,如果在安装过程中遇到问题,请参考Caffe官方文档进行解决。
阅读全文