Docker笔记--Ubuntu20.04基于Docker安装caffe-gpu

晓晓纳兰容若

已于 2023-03-20 10:32:07 修改

阅读量555

点赞数

分类专栏： Caffe Docker学习笔记文章标签： caffe docker 人工智能

于 2023-03-15 23:07:49 首次发布

本文链接：https://blog.csdn.net/weixin_43863869/article/details/129572303

版权

10 篇文章 2 订阅

订阅专栏

1 篇文章 0 订阅

订阅专栏

前言

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

sudo docker search caffe

nvidia-docker pull bvlc/caffe:gpu

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

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

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

① 测试 caffe 库

python

import caffe

① 使用镜像创建容器时可能会出现以下错误：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

关注