ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

最全caffe安装踩坑记录(Anaconda,nvidia-docker,linux编译)

2019-07-20 14:55:17  阅读:412  来源: 互联网

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



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

 

  

      



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

专注分享技术,共同学习,共同进步。

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有