/
Dockerfile
105 lines (92 loc) · 3.96 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
# ==================================================================
# module list
# ------------------------------------------------------------------
# python 3.8 (apt)
# pytorch latest (pip)
# ==================================================================
# ===============通过本地文件Dockerfile方式构建镜像===================
# sudo docker build -t guyu_cu100:v1 .
# ===============远端拉取方式构建====================================
# docker pull nvidia/cuda:10.0-devel-ubuntu18.04
# docker pull nvidia/cuda:10.2-devel-ubuntu18.04
# ===========create new container by docker images==================
# docker run -itd --runtime=nvidia --gpus=all -e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video,graphics -v /root/Documents/guyu:/data/ --name dmrnet --privileged=true -v /dev/shm:/dev/shm a411307c4f6b /bin/bash
# -it必备 d表示在背景里运行; -v 磁盘挂载 本地绝对路径:容器的绝对路径; --name 给你的container取个响亮的芳名;-v /dev/shm:/dev/shm 共享内存; a411307c4f6b image ID; /bin/bash 执行命令; -p 端口映射
# 他会创建一个容器出来,本身带着id,然后docker ps -a 即可查看
# =====然后我们进入容器 docker exec -it container-id (f34d733ff398) /bin/bash
# =====退出容器 ctrl+D
# =====启动 docker, docker start (container-id) ,停止容器 docker stop (container-id)
# =====删除镜像, docker rmi image-id, 删除container docker rm container-id
# ============想让它一直运行的话,可以使用disown命令或sh ./start.sh======
# docker export container-id > name.tar 容器导出,带着它到处跑,直接导出的是没有压缩过的,你可以自己压缩一下
# docker import name.tar 容器导入
FROM nvidia/cuda:10.0-devel-ubuntu18.04
ENV LANG C.UTF-8
RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
PIP_INSTALL="python -m pip --no-cache-dir install --upgrade" && \
GIT_CLONE="git clone --depth 10" && \
rm -rf /var/lib/apt/lists/* \
/etc/apt/sources.list.d/cuda.list \
/etc/apt/sources.list.d/nvidia-ml.list && \
apt-get update && \
# ==================================================================
# tools
# ------------------------------------------------------------------
DEBIAN_FRONTEND=noninteractive $APT_INSTALL \
build-essential \
apt-utils \
ca-certificates \
wget \
git \
vim \
libssl-dev \
curl \
unzip \
unrar \
cmake \
&& \
# ==================================================================
# python
# ------------------------------------------------------------------
apt-get update && \
DEBIAN_FRONTEND=noninteractive $APT_INSTALL \
python3.8 \
python3.8-dev \
python3.8-distutils \
&& \
wget -O ~/get-pip.py https://bootstrap.pypa.io/get-pip.py && \
python3.8 ~/get-pip.py && \
ln -s /usr/bin/python3.8 /usr/local/bin/python && \
$PIP_INSTALL \
numpy \
scipy \
pandas \
scikit-image \
scikit-learn \
matplotlib \
Cython \
tqdm \
&& \
# ==================================================================
# pytorch
# ------------------------------------------------------------------
#pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio==0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
$PIP_INSTALL \
future \
numpy \
protobuf \
enum34 \
pyyaml \
typing \
&& \
$PIP_INSTALL \
--pre torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f \
https://download.pytorch.org/whl/torch_stable.html \
&& \
# ==================================================================
# config & cleanup
# ------------------------------------------------------------------
ldconfig && \
apt-get clean && \
apt-get autoremove && \
rm -rf /var/lib/apt/lists/* /tmp/* ~/*