【Linux】Dockerfile 创建centos下pytorch程序运行环境的镜像

最新推荐文章于 2024-08-01 18:51:42 发布

哈士奇的布偶

最新推荐文章于 2024-08-01 18:51:42 发布

阅读量1.9k

点赞数

分类专栏： linux 文章标签： Docker

本文链接：https://blog.csdn.net/bpp94_github/article/details/83832255

版权

本文介绍了如何通过Dockerfile创建一个包含CentOS、CUDA 8.0、cudnn 7、ffmpeg、Anaconda3.4、cv2和Pytorch0.4.1的镜像，用于运行基于PyTorch的卷积神经网络程序。详细步骤包括编写Dockerfile、启动Docker服务、构建和发布镜像，并提供了管理Docker镜像的建议。

摘要由CSDN通过智能技术生成

目的：

基于pytorch框架实现了一个卷积神经网络，现想创建该程序的运行镜像。

程序运行所需要的环境：

Centos

CUDA 8.0

cudnn 7

ffmpeg

Anaconda3.4

cv2

Pytorch0.4.1

流程：

0. 我的程序在test文件夹下，因此先进入test文件夹所在的路径

1. 编写Dockerfile

2. service docker start 开启Docker服务

3. docker build 指令创建镜像（docker build -t imageName:tag .）

4. docker tag 给本地镜像打标签（docker tag 本地镜像名：标签远程仓库名/镜像名/标签）

4. docker push 发布镜像（登陆仓库之后）（docker push 远程仓库名/镜像名/标签）

5. service docker stop 停止Docker服务

Dockerfile:

FROM nvidia/cuda:8.0-cudnn7-devel-centos7 #  基础镜像
LABEL maintainer "bbp94" #维护者信息

# 安装bzip2
RUN yum -y install bzip2

# 安装ffmpeg
RUN yum install -y epel-release && \
    rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-EPEL-7 && \
    rpm --import