目录

前言

1--搜索caffe镜像

2--使用Nvidia Docker安装caffe-gpu

3--使用镜像创建容器

4--测试

5--相关报错总结

---

前言

        在主机从头配置 caffe 会出现许多环境兼容问题，这里使用 Docker 安装 GPU 版本的深度学习框架 caffe；在配置caffe前，需要安装 docker 和 Nvidia Docker，可参考之前的笔记：

Docker笔记--Ubuntu20.04安装Docker

Docker笔记--Ubuntu20.04安装Nvidia-Docker

1--搜索caffe镜像

sudo docker search caffe

2--使用Nvidia Docker安装caffe-gpu

nvidia-docker pull bvlc/caffe:gpu

3--使用镜像创建容器

        关于容器的具体操作可参考之前的笔记：docker中容器的基本操作

sudo nvidia-docker run -it --name=caffe_gpu bvlc/caffe:gpu /bin/bash

        运行上述指令，可以创建一个名称为 caffe_gpu 的容器；

4--测试

① 测试 caffe 库

python

import caffe

5--相关报错总结

① 使用镜像创建容器时可能会出现以下错误：docker: Error response from daemon: Unknown runtime specified nvidia. See 'docker run --help'.

        解决方法：修改 daemon.json:

sudo gedit /etc/docker/daemon.json

         添加以下内容：

    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }

         重启 daemon 和 docker：

sudo systemctl daemon-reload

sudo systemctl restart docker