devel

huggingface · Sep 14, 2024 · 0c1c6bd · 0c1c6bd
1 parent 87c9f3e
commit 0c1c6bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_onnxruntime_train.yml b/.github/workflows/test_onnxruntime_train.yml
@@ -13,7 +13,7 @@ jobs:
       group: aws-g6-4xlarge-plus
 
     container:
-      image: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+      image: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
       options: --gpus all
 
     steps: