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

相关内容

热门资讯

分享实测“丽水跑得快其实有挂是... 您好:丽水跑得快这款游戏可以开挂,确实是有挂的,需要了解加客服微信【5951795】很多玩家在丽水跑...
重磅消息“动力乐园,可以开挂吗... 您好:动力乐园,这款游戏可以开挂,确实是有挂的,需要了解加客服微信【6670747】很多玩家在这款游...
[实测教程]“掌酷天天开心有没... [实测教程]“掌酷天天开心有没有挂”!确实真的有挂您好:掌酷天天开心这款游戏可以开挂,确实是有挂的,...
玩家实测“红豆娱乐透视挂辅助软... 您好:红豆娱乐这款游戏可以开挂,确实是有挂的,需要软件加微信【4194432】,很多玩家在红豆娱乐这...
给你科普“大宝苏北麻将到底有没... 您好:【大宝苏北麻将】这款游戏可以开挂,确实是有挂的,需要了解加客服微信【6948699】很多玩家在...