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. 下载软件包

   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

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

   1. #安装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/* 

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