Skip to content

GitLab

Unverified Commit 3df4f8cb authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI] add pyg into CI image (#6997)

parent dfa32286
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 9 deletions
Jenkinsfile
View file @ 3df4f8cb
...@@ -318,7 +318,7 @@ pipeline { ...@@ -318,7 +318,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700" image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -337,7 +337,7 @@ pipeline { ...@@ -337,7 +337,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700" image "dgllib/dgl-ci-gpu:cu116_v240123_1000"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -392,7 +392,7 @@ pipeline { ...@@ -392,7 +392,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700" image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -411,7 +411,7 @@ pipeline { ...@@ -411,7 +411,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700" image "dgllib/dgl-ci-gpu:cu116_v240123_1000"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -464,7 +464,7 @@ pipeline { ...@@ -464,7 +464,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700" image "dgllib/dgl-ci-gpu:cu116_v240123_1000"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -489,7 +489,7 @@ pipeline { ...@@ -489,7 +489,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700" image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root --shm-size=4gb" args "-u root --shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -542,7 +542,7 @@ pipeline { ...@@ -542,7 +542,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700" image "dgllib/dgl-ci-gpu:cu116_v240123_1000"
args "-u root --runtime nvidia --shm-size=8gb" args "-u root --runtime nvidia --shm-size=8gb"
alwaysPull true alwaysPull true
} }
...@@ -571,7 +571,7 @@ pipeline { ...@@ -571,7 +571,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700" image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root --shm-size=4gb" args "-u root --shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -618,7 +618,7 @@ pipeline { ...@@ -618,7 +618,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700" image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ 3df4f8cb
...@@ -21,4 +21,5 @@ torch==1.13.0+cpu ...@@ -21,4 +21,5 @@ torch==1.13.0+cpu
torchdata torchdata
torcheval torcheval
torchmetrics torchmetrics
torch_geometric
tqdm tqdm
docker/install/conda_env/torch_gpu_pip.txt
View file @ 3df4f8cb
...@@ -19,4 +19,5 @@ torch==1.13.0+cu116 ...@@ -19,4 +19,5 @@ torch==1.13.0+cu116
torchdata torchdata
torcheval torcheval
torchmetrics torchmetrics
torch_geometric
tqdm tqdm
script/dgl_dev.yml.template
View file @ 3df4f8cb
...@@ -30,6 +30,7 @@ dependencies: ...@@ -30,6 +30,7 @@ dependencies:
- torchdata>=0.5.0 - torchdata>=0.5.0
- torcheval - torcheval
- torchmetrics - torchmetrics
- torch_geometric
- tqdm - tqdm
- boto3 # AWS SDK for python - boto3 # AWS SDK for python
- sphinx==4.2.0 - sphinx==4.2.0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment