Skip to content

GitLab

Unverified Commit 6dcdaf59 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI] add torcheval into CI docker (#6527)

parent 9111deee
Show whitespace changes
Inline Side-by-side
Showing with 11 additions and 9 deletions
Jenkinsfile
View file @ 6dcdaf59
...@@ -317,7 +317,7 @@ pipeline { ...@@ -317,7 +317,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900" image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -336,7 +336,7 @@ pipeline { ...@@ -336,7 +336,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900" image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -391,7 +391,7 @@ pipeline { ...@@ -391,7 +391,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900" image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -410,7 +410,7 @@ pipeline { ...@@ -410,7 +410,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900" image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -463,7 +463,7 @@ pipeline { ...@@ -463,7 +463,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900" image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -488,7 +488,7 @@ pipeline { ...@@ -488,7 +488,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900" image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root --shm-size=4gb" args "-u root --shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -541,7 +541,7 @@ pipeline { ...@@ -541,7 +541,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900" image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
args "-u root --runtime nvidia --shm-size=8gb" args "-u root --runtime nvidia --shm-size=8gb"
alwaysPull true alwaysPull true
} }
...@@ -570,7 +570,7 @@ pipeline { ...@@ -570,7 +570,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900" image "dgllib/dgl-ci-cpu:v231103_1700"
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:v231027_1900" image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ 6dcdaf59
...@@ -19,5 +19,6 @@ scikit-learn ...@@ -19,5 +19,6 @@ scikit-learn
scipy scipy
torch==1.13.0+cpu torch==1.13.0+cpu
torchdata torchdata
torcheval
torchmetrics torchmetrics
tqdm tqdm
docker/install/conda_env/torch_gpu_pip.txt
View file @ 6dcdaf59
...@@ -17,5 +17,6 @@ scikit-learn ...@@ -17,5 +17,6 @@ scikit-learn
scipy scipy
torch==1.13.0+cu116 torch==1.13.0+cu116
torchdata torchdata
torcheval
torchmetrics torchmetrics
tqdm tqdm
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