目录
前言
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
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 库
pythonimport 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-reloadsudo systemctl restart docker