Skip to content

GitLab

You need to sign in or sign up before continuing.
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 {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700"
image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root"
alwaysPull true
}
......@@ -337,7 +337,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
image "dgllib/dgl-ci-gpu:cu116_v240123_1000"
args "-u root"
alwaysPull true
}
......@@ -392,7 +392,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700"
image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root"
alwaysPull true
}
......@@ -411,7 +411,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -464,7 +464,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -489,7 +489,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -542,7 +542,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -571,7 +571,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -618,7 +618,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231103_1700"
image "dgllib/dgl-ci-cpu:v240123_1000"
args "-u root"
alwaysPull true
}
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ 3df4f8cb
......@@ -21,4 +21,5 @@ torch==1.13.0+cpu
torchdata
torcheval
torchmetrics
torch_geometric
tqdm
docker/install/conda_env/torch_gpu_pip.txt
View file @ 3df4f8cb
......@@ -19,4 +19,5 @@ torch==1.13.0+cu116
torchdata
torcheval
torchmetrics
torch_geometric
tqdm
script/dgl_dev.yml.template
View file @ 3df4f8cb
......@@ -30,6 +30,7 @@ dependencies:
- torchdata>=0.5.0
- torcheval
- torchmetrics
- torch_geometric
- tqdm
- boto3 # AWS SDK for python
- 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