Skip to content

Commit 80dd75a

Browse files
haobibohaobibo
authored
add torch-cu102 build (#64)
1 parent 6bf42c0 commit 80dd75a

File tree

3 files changed

+23
-23
lines changed

3 files changed

+23
-23
lines changed

‎.github/workflows/docker.yml

+22-17
Original file line numberDiff line numberDiff line change
@@ -83,12 +83,10 @@ jobs:
8383
- run: |
8484
source ./tool.sh
8585
build_image core latest docker_core/Dockerfile \
86-
--build-arg "ARG_PROFILE_PYTHON=base,datascience,database,nlp,cv,bio,chem,tf2,torch" \
86+
--build-arg "ARG_PROFILE_PYTHON=base,datascience,database,nlp,cv,chem,tf2,torch" \
8787
--build-arg "ARG_PROFILE_R=base,datascience,rstudio,rshiny" \
8888
--build-arg "ARG_PROFILE_NODEJS=base" \
8989
--build-arg "ARG_PROFILE_JAVA=base,maven" \
90-
--build-arg "ARG_PROFILE_GO=base" \
91-
--build-arg "ARG_PROFILE_JULIA=base" \
9290
--build-arg "ARG_PROFILE_LATEX=base,cjk"
9391
push_image
9492
@@ -116,14 +114,6 @@ jobs:
116114
- uses: actions/checkout@v2
117115
- run: source ./tool.sh && build_image py-cv latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,cv" && push_image
118116

119-
qpod_py-bio:
120-
name: qpod/py-bio
121-
needs: qpod_base
122-
runs-on: ubuntu-latest
123-
steps:
124-
- uses: actions/checkout@v2
125-
- run: source ./tool.sh && build_image py-bio latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,bio" && push_image
126-
127117
qpod_py-chem:
128118
name: qpod/py-chem
129119
needs: qpod_base
@@ -138,15 +128,15 @@ jobs:
138128
runs-on: ubuntu-latest
139129
steps:
140130
- uses: actions/checkout@v2
141-
- run: source ./tool.sh && build_image py-std latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,database,nlp,cv,bio,chem" && push_image
131+
- run: source ./tool.sh && build_image py-std latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,database,nlp,cv,chem" && push_image
142132

143133
qpod_py-jdk:
144134
name: qpod/py-jdk
145135
needs: qpod_base
146136
runs-on: ubuntu-latest
147137
steps:
148138
- uses: actions/checkout@v2
149-
- run: source ./tool.sh && build_image py-jdk latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,database,nlp,cv,bio,chem" --build-arg "ARG_PROFILE_JAVA=base" && push_image
139+
- run: source ./tool.sh && build_image py-jdk latest docker_core/Dockerfile --build-arg "ARG_PROFILE_PYTHON=datascience,database,nlp,cv,chem" --build-arg "ARG_PROFILE_JAVA=base" && push_image
150140

151141
qpod_r-base:
152142
name: qpod/r-base
@@ -250,8 +240,8 @@ jobs:
250240
--build-arg "BASE_IMG=cuda_11.2"
251241
alias_image py-cuda-11.2 latest tf2 latest && push_image
252242
253-
qpod_torch:
254-
name: qpod/torch
243+
qpod_torch_cu113:
244+
name: qpod/torch-cuda113
255245
needs: qpod_cuda_11_3
256246
runs-on: ubuntu-latest
257247
steps:
@@ -261,7 +251,22 @@ jobs:
261251
build_image py-cuda-11.3 latest docker_core/Dockerfile \
262252
--build-arg "ARG_PROFILE_PYTHON=torch" \
263253
--build-arg "BASE_IMG=cuda_11.3"
264-
alias_image py-cuda-11.3 latest torch latest && push_image
254+
alias_image py-cuda-11.3 latest torch-cu113 latest
255+
alias_image py-cuda-11.3 latest torch latest
256+
push_image
257+
258+
qpod_torch_cu102:
259+
name: qpod/torch
260+
needs: qpod_cuda_10_2
261+
runs-on: ubuntu-latest
262+
steps:
263+
- uses: actions/checkout@v2
264+
- run: |
265+
source ./tool.sh
266+
build_image py-cuda-10.2 latest docker_core/Dockerfile \
267+
--build-arg "ARG_PROFILE_PYTHON=torch" \
268+
--build-arg "BASE_IMG=cuda_10.2"
269+
alias_image py-cuda-10.2 latest torch-cuda102 latest && push_image
265270
266271
qpod_core-cuda:
267272
name: qpod/core-cuda
@@ -273,7 +278,7 @@ jobs:
273278
source ./tool.sh
274279
build_image full-cuda-11.3 latest docker_core/Dockerfile \
275280
--build-arg "BASE_IMG=cuda_11.3" \
276-
--build-arg "ARG_PROFILE_PYTHON=base,datascience,database,nlp,cv,bio,chem,tf2,torch" \
281+
--build-arg "ARG_PROFILE_PYTHON=base,datascience,database,nlp,cv,chem,tf2,torch" \
277282
--build-arg "ARG_PROFILE_R=base,datascience,rstudio,rshiny" \
278283
--build-arg "ARG_PROFILE_NODEJS=base" \
279284
--build-arg "ARG_PROFILE_JAVA=base,maven" \

‎README.md

+1-2
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,6 @@ Typically, you can choose `full` / `full-cuda` if you have enough disk space and
6565
| py-data | https://hub.docker.com/r/qpod/py-data | qpod/base | Python environment customized for Data Science tasks. |
6666
| py-nlp | https://hub.docker.com/r/qpod/py-nlp | qpod/base | Python environment customized for NLP tasks. |
6767
| py-cv | https://hub.docker.com/r/qpod/py-cv | qpod/base | Python environment customized for Computer Vision tasks. |
68-
| py-bio | https://hub.docker.com/r/qpod/py-bio | qpod/base | Python environment customized for Bioinfo tasks. |
6968
| py-chem | https://hub.docker.com/r/qpod/py-chem | qpod/base | Python environment customized for Computational Chemistry tasks. |
7069
| py-std | https://hub.docker.com/r/qpod/py-std | qpod/base | Python environment including all the packages mentioned above installed. |
7170
| py-jdk | https://hub.docker.com/r/qpod/py-jdk | qpod/base | `py-std` plus OpenJDK. (no LaTex) |
@@ -78,7 +77,7 @@ Typically, you can choose `full` / `full-cuda` if you have enough disk space and
7877
| go | https://hub.docker.com/r/qpod/go | qpod/base | Minimal Golang environment. |
7978
| julia | https://hub.docker.com/r/qpod/julia | qpod/base | Minimal Julia environment. |
8079
| octave | https://hub.docker.com/r/qpod/octave | qpod/base | Minimal Octave environment + LaTex. |
81-
| core | https://hub.docker.com/r/qpod/core | qpod/base | ➕ Full Python environment (data + nlp + cv + bio + chem + tensorflow + pytorch)<br/> ➕ Full R environment (datascience + RStudio + RShiny) + LaTex <br/> ➕ Base NodeJS environment <br/> ➕ Base Java environment (OpenJDK + maven) <br/> ➕ Minimal Golang environment <br/> ➕ Minimal Julia environment <br/> ➕ Minimal Octave environment |
80+
| core | https://hub.docker.com/r/qpod/core | qpod/base | ➕ Full Python environment (data + nlp + cv + chem + tensorflow + pytorch)<br/> ➕ Full R environment (datascience + RStudio + RShiny) + LaTex <br/> ➕ Base NodeJS environment <br/> ➕ Base Java environment (OpenJDK + maven) <br/> ➕ Minimal Golang environment <br/> ➕ Minimal Julia environment <br/> ➕ Minimal Octave environment |
8281
| core-dev, full | https://hub.docker.com/r/qpod/core-dev | qpod/core | All features and packages (Python, R, RStudio, OpenJDK, NodeJS, Go, Julia, LaTex) ➕ IDE tools: JupyterLab / Jupyter Notebook, VSCode Server |
8382

8483
</details>

‎docker_core/work/install_list_PY_bio.pip

-4
This file was deleted.

0 commit comments

Comments
 (0)