多种方式安装caffe
1.Anaconda安装caffe
1.首先安装anaconda
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 <容器id> # 已打开的容器
# 退出容器 ctrl + d (容器不运行)或者ctrl q+p(容器后台运行)
# 进入未运行的容器
docker ps -a # 查看容器id
docker start <id>
docker attach <id>
# 容器改名
dockers rename 原id 新id
4.更新源
apt uptate
5.测试
python import caffe