[系统运维] docker-build镜像(ubuntu18.04+python3.8+torch+cuda)

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 系统运维 -> docker-build镜像(ubuntu18.04+python3.8+torch+cuda) -> 正文阅读

[系统运维]docker-build镜像(ubuntu18.04+python3.8+torch+cuda)

从官网下载需要的软件包版本https://download.pytorch.org/whl/lts/1.8/torch_lts.htmlhttps://download.pytorch.org/whl/lts/1.8/torch_lts.html

下载软件包
1. torch-1.8.2+cpu-cp38-cp38-linux_x86_64.whl
  
  torch-1.8.2+cu111-cp38-cp38-linux_x86_64.whl
  torchaudio-0.8.2-cp38-cp38-linux_x86_64.whl
  torchvision-0.9.2+cu111-cp38-cp38-linux_x86_64.whl

创建Dockerfile文件--touch Dockerfile? 然后填写下方内容

#安装python运行环境
#
################################################

#基于哪个镜像生成新的镜像
FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04

RUN rm /etc/apt/sources.list.d/cuda.list


#作者名
MAINTAINER SunPengfei

#设置环境变量
ENV TZ Asia/Shanghai
ENV LANG zh_CN.UTF-8
# 拷贝下载好的whl文件到镜像中
COPY torch-1.8.2+cpu-cp38-cp38-linux_x86_64.whl /tmp
COPY torch-1.8.2+cu111-cp38-cp38-linux_x86_64.whl /tmp
COPY torchaudio-0.8.2-cp38-cp38-linux_x86_64.whl /tmp
COPY torchvision-0.9.2+cu111-cp38-cp38-linux_x86_64.whl /tmp


#执行命令
#替换为阿里源
RUN sed -i 's#http://archive.ubuntu.com/#http://mirrors.aliyun.com/#' /etc/apt/sources.list \
    && sed -i 's#http://security.ubuntu.com/#http://mirrors.aliyun.com/#' /etc/apt/sources.list

#更新软件源并安装软件
RUN apt-get update -y \
    && apt-get -y install iputils-ping \
    && apt-get -y install wget \
    && apt-get -y install net-tools \
    && apt-get -y install vim \
    && apt-get -y install openssh-server \
    && apt-get -y install python3.8 \
    && apt-get -y install python3-pip python3-dev \
    && cd /usr/local/bin \
    && rm -f python \
    && rm -f python3 \
    && rm -f pip \
    && rm -f pip3 \
    && ln -s /usr/bin/python3.8 python \
    && ln -s /usr/bin/python3.8 python3 \
    && ln -s /usr/bin/pip3 pip \
    && ln -s /usr/bin/pip3 pip3 \
    && python -m pip install --upgrade pip \
	&& cd /tmp \
	&& pip3 install torch-1.8.2+cpu-cp38-cp38-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
	&& pip3 install torch-1.8.2+cu111-cp38-cp38-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
	&& pip3 install torchaudio-0.8.2-cp38-cp38-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
	&& pip3 install torchvision-0.9.2+cu111-cp38-cp38-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
    && apt-get clean \
    && rm -rf /tmp/* /var/lib/apt/lists/* /var/tmp/* \

执行dockerfile脚本
1. docker build -t zgh_ir_analy/real_time:v1 .? ?#??注释docker build -t（镜像tag）工作路径
完成之后创建镜像，继续安装你想装的环境，收尾
1. docker run --name container_name -i -t -d tag
2. 进入镜像?docker exec -it container_id /bin/bash
保存更改后的镜像
1. docker commit container_id?tag
  ?