Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "bc2ba004c60aaaa2b16c2b43b98fdd6c10098300"
Unverified Commit e6e54304 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI]upgrade pytorch to 1.13+cpu_cu116 (#5977)

parent 4b456507
Hide whitespace changes
Inline Side-by-side
Showing with 17 additions and 15 deletions
Jenkinsfile
View file @ e6e54304
...@@ -287,7 +287,7 @@ pipeline { ...@@ -287,7 +287,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -305,7 +305,7 @@ pipeline { ...@@ -305,7 +305,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606" image "dgllib/dgl-ci-gpu:cu116_v230711"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -360,7 +360,7 @@ pipeline { ...@@ -360,7 +360,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true alwaysPull true
} }
} }
...@@ -377,7 +377,7 @@ pipeline { ...@@ -377,7 +377,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606" image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia" args "--runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -406,7 +406,7 @@ pipeline { ...@@ -406,7 +406,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true alwaysPull true
} }
} }
...@@ -427,7 +427,7 @@ pipeline { ...@@ -427,7 +427,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v230210" image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia" args "--runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -437,6 +437,8 @@ pipeline { ...@@ -437,6 +437,8 @@ pipeline {
steps { steps {
unit_test_linux('tensorflow', 'gpu') unit_test_linux('tensorflow', 'gpu')
} }
// Tensorflow does not support cuda 11.6 yet.
when { expression { false } }
} }
} }
post { post {
...@@ -449,7 +451,7 @@ pipeline { ...@@ -449,7 +451,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
args "--shm-size=4gb" args "--shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -501,7 +503,7 @@ pipeline { ...@@ -501,7 +503,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606" image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia --shm-size=8gb" args "--runtime nvidia --shm-size=8gb"
alwaysPull true alwaysPull true
} }
...@@ -529,7 +531,7 @@ pipeline { ...@@ -529,7 +531,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
args "--shm-size=4gb" args "--shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -575,7 +577,7 @@ pipeline { ...@@ -575,7 +577,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606" image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true alwaysPull true
} }
} }
......
docker/Dockerfile.ci_gpu
View file @ e6e54304
# CI docker GPU env # CI docker GPU env
FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu18.04
ENV TZ=US ENV TZ=US
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
......
docker/install/conda_env/torch_cpu.yml
View file @ e6e54304
name: pytorch-ci name: pytorch-ci
dependencies: dependencies:
- python=3.7.0 - python=3.8
- pip - pip
- pip: - pip:
- --find-links https://download.pytorch.org/whl/torch_stable.html - --find-links https://download.pytorch.org/whl/torch_stable.html
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ e6e54304
...@@ -17,7 +17,7 @@ rdflib ...@@ -17,7 +17,7 @@ rdflib
requests[security]==2.28 requests[security]==2.28
scikit-learn scikit-learn
scipy scipy
torch==1.12.0+cpu torch==1.13.0+cpu
torchdata torchdata
torchmetrics torchmetrics
tqdm tqdm
docker/install/conda_env/torch_gpu.yml
View file @ e6e54304
name: pytorch-ci name: pytorch-ci
dependencies: dependencies:
- python=3.7.0 - python=3.8
- pip - pip
- pip: - pip:
- --find-links https://download.pytorch.org/whl/torch_stable.html - --find-links https://download.pytorch.org/whl/torch_stable.html
......
docker/install/conda_env/torch_gpu_pip.txt
View file @ e6e54304
...@@ -15,7 +15,7 @@ rdflib ...@@ -15,7 +15,7 @@ rdflib
requests[security]==2.28 requests[security]==2.28
scikit-learn scikit-learn
scipy scipy
torch==1.12.0+cu102 torch==1.13.0+cu116
torchdata torchdata
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