Skip to content

GitLab

Unverified Commit e424d296 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

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

parent 539335ce
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
docker/install/conda_env/torch_cpu_pip.txt
View file @ e424d296
--find-links https://download.pytorch.org/whl/torch_stable.html --find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cpu torch==1.9.0+cpu
pytest pytest
nose nose
numpy numpy
......
docker/install/conda_env/torch_gpu_pip.txt
View file @ e424d296
--find-links https://download.pytorch.org/whl/torch_stable.html --find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cu101 torch==1.9.0+cu102
pytest pytest
nose nose
numpy numpy
......
docker/pods/ci-cpu.yaml
View file @ e424d296
...@@ -5,7 +5,7 @@ spec: ...@@ -5,7 +5,7 @@ spec:
runAsUser: 0 runAsUser: 0
containers: containers:
- name: dgl-ci-cpu - name: dgl-ci-cpu
image: dgllib/dgl-ci-cpu:cu101_v220123 image: dgllib/dgl-ci-cpu:cu101_v220217
imagePullPolicy: Always imagePullPolicy: Always
tty: true tty: true
resources: resources:
......
docker/pods/ci-gpu.yaml
View file @ e424d296
...@@ -5,7 +5,7 @@ spec: ...@@ -5,7 +5,7 @@ spec:
runAsUser: 0 runAsUser: 0
containers: containers:
- name: dgl-ci-gpu - name: dgl-ci-gpu
image: dgllib/dgl-ci-gpu:cu101_v220123 image: dgllib/dgl-ci-gpu:cu101_v220217
imagePullPolicy: Always imagePullPolicy: Always
tty: true tty: true
resources: resources:
......
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