Add CUDA-enabled PyTorch and TensorFlow images

osg-htc · Jun 18, 2024 · 340cd61 · 340cd61
1 parent 70270fa
commit 340cd61
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -28,6 +28,8 @@ jobs:
         image-name:
         - htc-minimal-notebook
         - htc-datascience-notebook
+        - htc-pytorch-notebook
+        - htc-tensorflow-notebook
 
     steps: