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

rudongyu · Nov 3, 2023 · 6dcdaf5 · 6dcdaf5
1 parent 9111dee
commit 6dcdaf5
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -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
                 }

diff --git a/docker/install/conda_env/torch_cpu_pip.txt b/docker/install/conda_env/torch_cpu_pip.txt
@@ -19,5 +19,6 @@ scikit-learn
 scipy
 torch==1.13.0+cpu
 torchdata
+torcheval
 torchmetrics
 tqdm
diff --git a/docker/install/conda_env/torch_gpu_pip.txt b/docker/install/conda_env/torch_gpu_pip.txt
@@ -17,5 +17,6 @@ scikit-learn
 scipy
 torch==1.13.0+cu116
 torchdata
+torcheval
 torchmetrics
 tqdm