From d6262c30d505cb53400b1baa2544dcce5a62eae3 Mon Sep 17 00:00:00 2001
From: QPod0 <QPod0@outlook.com>
Date: Sat, 3 Sep 2022 03:47:31 +0800
Subject: [PATCH] add pynlp-cuda102
---
.github/workflows/docker.yml | 19 ++++++++++++++++---
docker_core/work/install_list_PY_nlp.pip | 7 ++++---
2 files changed, 20 insertions(+), 6 deletions(-)
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index f3eb606..026c0a7 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -279,17 +279,30 @@ jobs:
push_image
- qpod_py-nlp:
- name: qpod/py-nlp
+ qpod_py-nlp-cuda116:
+ name: qpod/py-nlp-cuda116
needs: qpod_base
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- run: |
source ./tool.sh
- build_image py-nlp latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_11.6" --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,nlp"
+ build_image py-nlp-cuda116 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_11.6" --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,nlp"
+ alias_image py-nlp-cuda116 latest py-nlp latest
push_image
+ qpod_py-nlp-cuda102:
+ name: qpod/py-nlp-cuda102
+ needs: qpod_base
+ runs-on: ubuntu-latest
+ steps:
+ - uses: actions/checkout@v2
+ - run: |
+ source ./tool.sh
+ build_image py-nlp-cuda102 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_10.2" --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,nlp"
+ push_image
+
+
qpod_py-cv:
name: qpod/py-cv
needs: qpod_base
diff --git a/docker_core/work/install_list_PY_nlp.pip b/docker_core/work/install_list_PY_nlp.pip
index b1cbdfd..2d26a73 100644
--- a/docker_core/work/install_list_PY_nlp.pip
+++ b/docker_core/work/install_list_PY_nlp.pip
@@ -1,6 +1,7 @@
% This file contains python packages to be installed with pip line by line.
% Use percent char as line comment separator.
-pyparsing astroid nltk % Parsing, AST
-aiohttp==3.* % TEMP fix: req by datasets, 4.0 won't work
-transformers datasets % huggingface transformers
+pyparsing astroid % Parsing, AST
+aiohttp==3.* % TEMP fix: req by datasets, 4.0 won't work
+transformers tokenizers datasets % huggingface transformers
+rapidfuzz flashtext % string match