XM4HFl2RbY commited on
Commit
9bb4d7d
·
verified ·
1 Parent(s): 0ef06c8

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +45 -0
Dockerfile ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM ubuntu:latest
2
+
3
+ # 避免安装过程中出现交互提示
4
+ ENV DEBIAN_FRONTEND=noninteractive
5
+
6
+ # 安装 SSH 服务、Python 以及 webssh 所需的依赖
7
+ RUN apt-get update && \
8
+ apt-get install -y openssh-server python3 python3-pip && \
9
+ apt-get clean && \
10
+ rm -rf /var/lib/apt/lists/*
11
+
12
+ # 配置 SSH 服务
13
+ RUN mkdir /var/run/sshd
14
+ RUN echo 'PermitRootLogin yes' >> /etc/ssh/sshd_config
15
+ RUN echo 'PasswordAuthentication yes' >> /etc/ssh/sshd_config
16
+
17
+ # 安装 webssh (添加参数以适配较新版 Ubuntu 的 Python 环境限制)
18
+ RUN pip3 install webssh --break-system-packages
19
+
20
+ # 编写启动脚本,动态设置密码并同时启动 sshd 与 webssh
21
+ RUN echo '#!/bin/bash\n\
22
+ # 检查是否设置了密码环境变量\n\
23
+ if [ -z "$SSH_PASSWORD" ]; then\n\
24
+ echo "请在 Hugging Face 设置中配置 SSH_PASSWORD 环境变量"\n\
25
+ exit 1\n\
26
+ fi\n\
27
+ \n\
28
+ # 将环境变量的值赋予 root 用户作为登录密码\n\
29
+ echo "root:$SSH_PASSWORD" | chpasswd\n\
30
+ \n\
31
+ # 启动原生 SSH 服务\n\
32
+ /usr/sbin/sshd\n\
33
+ \n\
34
+ # 在前台启动 webssh,并绑定 Hugging Face 默认的 7860 端口\n\
35
+ wssh --address=0.0.0.0 --port=7860\n\
36
+ ' > /start.sh
37
+
38
+ # 赋予启动脚本执行权限
39
+ RUN chmod +x /start.sh
40
+
41
+ # 暴露 Hugging Face 要求的默认网络端口
42
+ EXPOSE 7860
43
+
44
+ # 设置容器启动时执行的命令
45
+ CMD ["/start.sh"]