Explorar o código

修改 dockerfile, 创建 python 环境

罗俊辉 hai 1 ano
pai
achega
6a3ccfa0a5
Modificáronse 3 ficheiros con 14 adicións e 12 borrados
  1. 12 6
      Dockerfile
  2. 2 2
      applications/model_init.py
  3. 0 4
      pressure_test.py

+ 12 - 6
Dockerfile

@@ -1,13 +1,19 @@
-# 使用官方Python镜像作为基础镜像
-FROM python:3.10
+FROM  registry.piaoquantv.com/piaoquan/python:3.10
 
-WORKDIR /app
+ENV LANG en_US.UTF-8
 
-# 复制依赖文件到容器中
-COPY requirements.txt /app/
+RUN apk update && apk add tzdata
+RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime
+RUN echo "Asia/Shanghai" > /etc/timezone
 
+ADD  lightgbm_score    /home/lightgbm_score
+WORKDIR /home/lightgbm-score
+
+# 升级 更新 pip, 使用清华源
 RUN pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
+# 安装依赖,使用清华源
 RUN pip install --no-cache-dir -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 
+EXPOSE 8000
 
-CMD ["python3"]
+CMD ["hypercorn","async_app:app","--config", "hypercorn.config.toml"]

+ 2 - 2
applications/model_init.py

@@ -22,9 +22,9 @@ class Models(object):
         self.label_encoder = LabelEncoder()
         print("标签分类器加载完成")
         print("开始加载模型")
-        self.model_v1 = lgb.Booster(model_file="/root/lightgbm_score/models/lightgbm_0409_all_tags.bin")
+        self.model_v1 = lgb.Booster(model_file="models/lightgbm_0409_all_tags.bin")
         print("模型 1 加载完成......")
-        self.model_v2 = lgb.Booster(model_file="/root/lightgbm_score/models/lightgbm_0409_spider.bin")
+        self.model_v2 = lgb.Booster(model_file="models/lightgbm_0409_spider.bin")
         print("模型 2 加载完成......")
         print("所有模型加载完成")
 

+ 0 - 4
pressure_test.py

@@ -32,10 +32,6 @@ def request_data(_url):
     print(res.json())
 
 
-    # sss = "请求: {}, 花费时间: {}, 请求状态码: {}, 请求结果: {}".format(index, e - t, res.status_code, res.text)
-    # print(sss)
-
-
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()  # 新建参数解释器对象
     parser.add_argument("--thread")