安装完cuda,cudnn后如何在anconda下安装caffe
时间: 2024-05-30 22:03:12 浏览: 111
安装 Caffe 需要进行一些环境配置,以下是在 Anaconda 环境下安装 Caffe 的步骤:
1. 安装依赖:
```
conda install protobuf-gpu leveldb snappy boost hdf5 opencv -c anaconda
conda install -c conda-forge gflags glog lmdb
```
2. 下载 Caffe 源码:
```
git clone https://github.com/BVLC/caffe.git
cd caffe
```
3. 修改 Makefile.config 文件:
```
cp Makefile.config.example Makefile.config
```
打开 Makefile.config 文件,按照注释进行修改,主要包括以下几个方面:
- 设置 CUDA 的路径:
```
CUDA_DIR := /usr/local/cuda
```
- 设置 cuDNN 的路径:
```
CUDNN_DIR := /usr/local/cudnn
```
- 设置 Python 的路径:
```
PYTHON_INCLUDE := /home/user/anaconda3/include \
/home/user/anaconda3/include/python3.6m \
/home/user/anaconda3/lib/python3.6/site-packages/numpy/core/include
```
- 设置 BLAS 的库文件:
```
BLAS := atlas
```
- 设置编译类型:
```
BUILD_TYPE := Release
```
4. 编译:
```
make all -j8
make test -j8
make runtest -j8
```
5. 安装 Python 接口:
```
make pycaffe -j8
```
6. 安装 Caffe:
```
make distribute
cd distribute
pip install -r requirements.txt
python setup.py install
```
安装完成后,可以通过 `import caffe` 来测试 Caffe 是否安装成功。
阅读全文