cuda-toolkit

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -96,6 +96,8 @@ jobs:
           python --version
           pip --version
           pip list
+          apt-get install -y cuda-toolkit
+          nvcc --version
         displayName: "Image info & NVIDIA"
 
       - bash: |

.azure/gpu-tests-pytorch.yml

@@ -100,6 +100,8 @@ jobs:
           python --version
           pip --version
           pip list
+          apt-get install -y cuda-toolkit
+          nvcc --version
         displayName: "Image info & NVIDIA"
 
       - bash: |