[CI] Use torch 1.9.0 in CI (dmlc#3752)

sanchit-misra · Feb 18, 2022 · e424d29 · e424d29
1 parent 539335c
commit e424d29
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/docker/install/conda_env/torch_cpu_pip.txt b/docker/install/conda_env/torch_cpu_pip.txt
@@ -1,5 +1,5 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.1+cpu
+torch==1.9.0+cpu
 pytest
 nose
 numpy

diff --git a/docker/install/conda_env/torch_gpu_pip.txt b/docker/install/conda_env/torch_gpu_pip.txt
@@ -1,5 +1,5 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.1+cu101
+torch==1.9.0+cu102
 pytest
 nose
 numpy

diff --git a/docker/pods/ci-cpu.yaml b/docker/pods/ci-cpu.yaml
@@ -5,7 +5,7 @@ spec:
     runAsUser: 0
   containers:
   - name: dgl-ci-cpu
-    image: dgllib/dgl-ci-cpu:cu101_v220123
+    image: dgllib/dgl-ci-cpu:cu101_v220217
     imagePullPolicy: Always
     tty: true
     resources:

diff --git a/docker/pods/ci-gpu.yaml b/docker/pods/ci-gpu.yaml
@@ -5,7 +5,7 @@ spec:
     runAsUser: 0
   containers:
     - name: dgl-ci-gpu
-      image: dgllib/dgl-ci-gpu:cu101_v220123
+      image: dgllib/dgl-ci-gpu:cu101_v220217
       imagePullPolicy: Always
       tty: true
       resources: