Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "86ecd4b795f865b5b615b8c54991c177bb3dbef5"
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
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 9 deletions
Jenkinsfile
View file @ 6dcdaf59
......@@ -317,7 +317,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900"
image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root"
alwaysPull true
}
......@@ -336,7 +336,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
image "dgllib/dgl-ci-gpu:cu116_v231103_1700"
args "-u root"
alwaysPull true
}
......@@ -391,7 +391,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900"
image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root"
alwaysPull true
}
......@@ -410,7 +410,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -463,7 +463,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -488,7 +488,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -541,7 +541,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -570,7 +570,7 @@ pipeline {
agent {
docker {
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"
alwaysPull true
}
......@@ -618,7 +618,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v231027_1900"
image "dgllib/dgl-ci-cpu:v231103_1700"
args "-u root"
alwaysPull true
}
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ 6dcdaf59
......@@ -19,5 +19,6 @@ scikit-learn
scipy
torch==1.13.0+cpu
torchdata
torcheval
torchmetrics
tqdm
docker/install/conda_env/torch_gpu_pip.txt
View file @ 6dcdaf59
......@@ -17,5 +17,6 @@ scikit-learn
scipy
torch==1.13.0+cu116
torchdata
torcheval
torchmetrics
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