Ubuntu 17.10下Caffe安装位置查询：Digits配置需指定其路径

你遇到的情况其实很常见——caffe-cuda本身是一个元包（metapackage），它的作用只是自动拉取安装Caffe GPU版本所需的依赖组件，自身并不包含核心的头文件、库或者二进制文件，所以用dpkg -L caffe-cuda只能看到少量文档文件。下面是具体的解决步骤：

1. 检查已安装的Caffe核心组件 #

先确认系统里是否已经安装了包含实际文件的Caffe包，执行以下命令：

dpkg -l | grep caffe

你应该能看到类似libcaffe-cuda（运行时库）和libcaffe-cuda-dev（开发用的头文件、静态库等）的条目。如果没有libcaffe-cuda-dev，说明你只装了运行环境，缺少开发所需的文件。

2. 安装Caffe开发包（如果缺失） #

如果上面的输出里没有libcaffe-cuda-dev，执行以下命令安装：

sudo apt-get update && sudo apt-get install libcaffe-cuda-dev

这个包会把Caffe的核心头文件、库文件等安装到系统标准路径下。

3. 查看完整的安装路径 #

安装完成后，用以下命令查看libcaffe-cuda-dev的所有文件位置：

dpkg -L libcaffe-cuda-dev

你会看到类似这些关键路径：

• 头文件：/usr/include/caffe
• 库文件：/usr/lib/x86_64-linux-gnu
• Python接口：/usr/lib/python3/dist-packages/caffe（或对应Python版本的路径）

4. 设置Digits所需的环境变量 #

Digits需要CAFFE_ROOT环境变量来定位Caffe的安装目录。对于apt安装的Caffe，我们可以把CAFFE_ROOT指向/usr（因为所有核心文件都在这个目录的子路径下）。

打开你的bash配置文件（比如~/.bashrc），添加以下内容：

export CAFFE_ROOT=/usr
export PYTHONPATH=$CAFFE_ROOT/lib/python3/dist-packages:$PYTHONPATH

（注意：如果你的Python版本是2.x，把路径里的python3改成python2）

保存文件后，执行以下命令让环境变量生效：

source ~/.bashrc

5. 验证设置 #

执行以下命令确认环境变量已正确设置：

echo $CAFFE_ROOT
python3 -c "import caffe; print('Caffe imported successfully')"

如果没有报错，说明Caffe的路径已经被正确识别，Digits应该可以正常使用了。

内容的提问来源于stack exchange，提问作者Chuck Carlson