atonyxu commited on
Commit
e8117ef
·
verified ·
1 Parent(s): f3c6bed

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -102
Dockerfile CHANGED
@@ -1,105 +1,12 @@
1
- FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
 
2
 
3
- ENV DEBIAN_FRONTEND=noninteractive \
4
- TZ=Europe/Paris
5
 
6
- # Remove any third-party apt sources to avoid issues with expiring keys.
7
- # Install some basic utilities
8
- RUN rm -f /etc/apt/sources.list.d/*.list && \
9
- apt-get update && apt-get install -y --no-install-recommends \
10
- curl \
11
- ca-certificates \
12
- sudo \
13
- git \
14
- wget \
15
- procps \
16
- git-lfs \
17
- zip \
18
- unzip \
19
- htop \
20
- vim \
21
- nano \
22
- bzip2 \
23
- libx11-6 \
24
- build-essential \
25
- libsndfile-dev \
26
- software-properties-common \
27
- && rm -rf /var/lib/apt/lists/*
28
 
29
- RUN add-apt-repository ppa:flexiondotorg/nvtop && \
30
- apt-get upgrade -y && \
31
- apt-get install -y --no-install-recommends nvtop
32
-
33
- RUN curl -sL https://deb.nodesource.com/setup_21.x | bash - && \
34
- apt-get install -y nodejs && \
35
- npm install -g configurable-http-proxy
36
-
37
- # Create a working directory
38
- WORKDIR /app
39
-
40
- # Create a non-root user and switch to it
41
- RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
42
- && chown -R user:user /app
43
- RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
44
- USER user
45
-
46
- # All users can use /home/user as their home directory
47
- ENV HOME=/home/user
48
- RUN mkdir $HOME/.cache $HOME/.config \
49
- && chmod -R 777 $HOME
50
-
51
- # Set up the Conda environment
52
- ENV CONDA_AUTO_UPDATE_CONDA=false \
53
- PATH=$HOME/miniconda/bin:$PATH
54
- RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh \
55
- && chmod +x ~/miniconda.sh \
56
- && ~/miniconda.sh -b -p ~/miniconda \
57
- && rm ~/miniconda.sh \
58
- && conda clean -ya
59
-
60
- WORKDIR $HOME/app
61
-
62
- #######################################
63
- # Start root user section
64
- #######################################
65
-
66
- USER root
67
-
68
- # User Debian packages
69
- ## Security warning : Potential user code executed as root (build time)
70
- RUN --mount=target=/root/packages.txt,source=packages.txt \
71
- apt-get update && \
72
- xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
73
- && rm -rf /var/lib/apt/lists/*
74
-
75
- RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
76
- bash /root/on_startup.sh
77
-
78
- RUN mkdir /data && chown user:user /data
79
-
80
- #######################################
81
- # End root user section
82
- #######################################
83
-
84
- USER user
85
-
86
- # Python packages
87
- RUN --mount=target=requirements.txt,source=requirements.txt \
88
- pip install --no-cache-dir --upgrade -r requirements.txt
89
-
90
- # Copy the current directory contents into the container at $HOME/app setting the owner to the user
91
- COPY --chown=user . $HOME/app
92
-
93
- RUN chmod +x start_server.sh
94
-
95
- COPY --chown=user login.html /home/user/miniconda/lib/python3.9/site-packages/jupyter_server/templates/login.html
96
-
97
- ENV PYTHONUNBUFFERED=1 \
98
- GRADIO_ALLOW_FLAGGING=never \
99
- GRADIO_NUM_PORTS=1 \
100
- GRADIO_SERVER_NAME=0.0.0.0 \
101
- GRADIO_THEME=huggingface \
102
- SYSTEM=spaces \
103
- SHELL=/bin/bash
104
-
105
- CMD ["./start_server.sh"]
 
1
+ # 使用官方基础镜像
2
+ FROM mltooling/ml-workspace:0.13.2
3
 
4
+ # 设置环境变量
5
+ ENV AUTHENTICATE_VIA_JUPYTER="mytoken"
6
 
7
+ # 暴露端口(这只是一个声明,实际端口映射是在运行时由 -p 参数控制)
8
+ EXPOSE 8080
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
+ # 如果有特定的命令或脚本需要在容器启动时运行,可以在这里设置
11
+ # CMD ENTRYPOINT 可以用来设置默认命令
12
+ # 注意,这不会自动处理卷挂载、容器名、共享内存大小或重启策略