Make numpy check optional (pytorch#153421)

Make numpy check optional (pytorch#149356)

We may want to skip numpy smoke tests. Hence making it optional

Pull Request resolved: pytorch#149356
Approved by: https://github.com/ZainRizvi

(cherry picked from commit 6e2b266)

Co-authored-by: atalman <[email protected]>

Author: pytorchbot

.ci/pytorch/smoke_test/smoke_test.py

@@ -76,10 +76,13 @@ def read_release_matrix():
 
 
 def test_numpy():
-    import numpy as np
+    try:
+        import numpy as np
 
-    x = np.arange(5)
-    torch.tensor(x)
+        x = np.arange(5)
+        torch.tensor(x)
+    except ImportError:
+        print("Numpy check skipped. Numpy is not installed.")
 
 
 def check_version(package: str) -> None:
@@ -410,6 +413,7 @@ def main() -> None:
     smoke_test_conv2d()
     test_linalg()
     test_numpy()
+
     if is_cuda_system:
         test_linalg("cuda")
         test_cuda_gds_errors_captured()