Docker笔记--Ubuntu20.04基于Docker安装caffe-gpu
创始人
2025-05-28 23:26:10

目录

前言

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 库

pythonimport 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-reloadsudo systemctl restart docker

相关内容

热门资讯

实测分享“天天十三水到底是不是... 您好:天天十三水这款游戏可以开挂,确实是有挂的,需要了解加客服微信【6355786】很多玩家在天天十...
分享实测“七千在线透视挂辅助器... 您好:七千在线这款游戏可以开挂,确实是有挂的,需要软件加微信【69174242】,很多玩家在七千在线...
分享实测“广西八一字牌到底可以... 您好:广西八一字牌这款游戏可以开挂,确实是有挂的,需要软件加微信【6355786】,很多玩家在广西八...
玩家必看“广西八一字牌其实有透... 您好:广西八一字牌这款游戏可以开挂,确实是有挂的,需要软件加微信【5902455】,很多玩家在广西八...
最新消息“我爱仪征麻将究竟有挂... 您好:我爱仪征麻将这款游戏可以开挂,确实是有挂的,需要软件加微信【4830828】很多玩家在这款游戏...