Unverified Commit 5833efe0 authored by Vibhu Jawa's avatar Vibhu Jawa Committed by GitHub
Browse files

Update cugraph_nightly_torch (#4757)

parent 9a72b78b
...@@ -298,7 +298,7 @@ pipeline { ...@@ -298,7 +298,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.10" image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.12"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -525,7 +525,7 @@ pipeline { ...@@ -525,7 +525,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" label "linux-gpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.10" image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.12"
args "--runtime nvidia --shm-size=8gb" args "--runtime nvidia --shm-size=8gb"
alwaysPull true alwaysPull true
} }
......
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