多种方式安装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