linux配置caffe环境,最全caffe安装踩坑记录（Anaconda,nvidia-docker,linux编译）

多种方式安装caffe

1.Anaconda安装caffe

2.创建虚拟环境(python2.7)

conda create -n caffe python=2.7 anaconda

3.安装caffe

# 然后下面二选一即可，安装caffe

conda install -c conda-forge caffe

conda install -c conda-forge/label/broken caffe

4.注意：

# 如果出现numpy导入错误，是因为 conda-forge中的numpy版本过低，可再输入下面命令

pip install numpy --upgrade

5.测试

# 测试是否成功

source activate caffe

python

import caffe

2.nvidia-docker安装caffe-gpu

注意：docker仅支持cpu

nvidia-docker支持GPU

1.安装nvidia-docker

wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb

sudo dpkg -i /tmp/nvidia-docker*.deb

# 测试

nvidia-docker run --rm nvidia/cuda nvidia-smi

2.拉取caffe-gpu镜像

nvidia-docker pull bvlc/caffe:gpu

3.启动一个容器

nvidia-docker run --rm -ti

docker attach # 已打开的容器

# 退出容器 ctrl + d (容器不运行)或者ctrl q+p(容器后台运行)

# 进入未运行的容器

docker ps -a # 查看容器id

docker start

docker attach

# 容器改名

dockers rename 原id 新id

4.更新源

apt uptate

5.测试

python

import caffe

标签：linux,容器,Anaconda,conda,nvidia,caffe,docker,安装

来源： https://www.cnblogs.com/houyong/p/11217688.html