update exception message of pytorchc cuda unit test

aws · Jul 9, 2024 · 7263d42 · 7263d42
1 parent ff89c2d
commit 7263d42
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/test/test_artifacts/v0/scripts/cuda_validation.py b/test/test_artifacts/v0/scripts/cuda_validation.py
@@ -40,4 +40,5 @@
 import torch
 
 if not torch.cuda.is_available():
-    raise Exception("Pytorch CUDA is not working in current environment. Make sure to execute this test case in GPU environment if you are not")
+    raise Exception("Pytorch is installed with CUDA support but not working in current environment. \
+                    Make sure to execute this test case in GPU environment if you are not")
diff --git a/test/test_artifacts/v1/scripts/cuda_validation.py b/test/test_artifacts/v1/scripts/cuda_validation.py
@@ -40,4 +40,5 @@
 import torch
 
 if not torch.cuda.is_available():
-    raise Exception("Pytorch CUDA is not working in current environment. Make sure to execute this test case in GPU environment if you are not")
+    raise Exception("Pytorch is installed with CUDA support but not working in current environment. \
+                    Make sure to execute this test case in GPU environment if you are not")
diff --git a/test/test_artifacts/v2/scripts/cuda_validation.py b/test/test_artifacts/v2/scripts/cuda_validation.py
@@ -40,4 +40,5 @@
 import torch
 
 if not torch.cuda.is_available():
-    raise Exception("Pytorch CUDA is not working in current environment. Make sure to execute this test case in GPU environment if you are not")
+    raise Exception("Pytorch is installed with CUDA support but not working in current environment. \
+                    Make sure to execute this test case in GPU environment if you are not")