Remove minor versions from device

CrayLabs · Sep 18, 2024 · 978b334 · 978b334
1 parent 745465d
commit 978b334
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 17 deletions.
diff --git a/smartsim/_core/_install/configs/mlpackages/LinuxX64CUDA118.json b/smartsim/_core/_install/configs/mlpackages/LinuxX64CUDA118.json
@@ -2,7 +2,7 @@
     "platform": {
         "operating_system":"linux",
         "architecture":"x86_64",
-        "device":"cuda-11.8"
+        "device":"cuda-11"
     },
     "ml_packages": [
         {
@@ -14,14 +14,14 @@
         },
         {
             "name": "libtorch",
-            "version": "2.4.0",
+            "version": "2.3.1",
             "pip_index": "https://download.pytorch.org/whl/cu118",
             "python_packages": [
-                "torch==2.4.0+cu118",
-                "torchvision==0.19.0+cu118",
-                "torchaudio==2.4.0+cu118"
+                "torch==2.3.1+cu118",
+                "torchvision==0.18.1+cu118",
+                "torchaudio==2.3.1+cu118"
             ],
-            "lib_source": "https://github.com/CrayLabs/ml_lib_builder/releases/download/v0.2/libtorch-2.4.0-linux-x64-cuda-11.8.0.tgz",
+            "lib_source": "https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu118.zip",
             "rai_patches": [
                 {
                     "description": "Patch RedisAI module to require C++17 standard instead of C++14",
@@ -50,7 +50,7 @@
                 "scikit-learn",
                 "onnxmltools"
             ],
-            "lib_source": "https://github.com/CrayLabs/ml_lib_builder/releases/download/v0.2/onnxruntime-1.17.3-linux-x64-cuda-11.8.0.tgz"
+            "lib_source": "https://github.com/microsoft/onnxruntime/releases/download/v1.17.3/onnxruntime-linux-x64-gpu-1.17.3.tgz"
         }
     ]
 }
diff --git a/smartsim/_core/_install/configs/mlpackages/LinuxX64ROCM57.json b/smartsim/_core/_install/configs/mlpackages/LinuxX64ROCM57.json
@@ -2,7 +2,7 @@
     "platform": {
         "operating_system":"linux",
         "architecture":"x86_64",
-        "device":"rocm-5.7"
+        "device":"rocm-5"
     },
     "ml_packages": [
         {
@@ -31,15 +31,6 @@
                 }
             ]
         },
-        {
-            "name": "libtensorflow",
-            "version": "2.13",
-            "pip_index": "",
-            "python_packages": [
-                "tensorflow-rocm==2.13.0.570"
-            ],
-            "lib_source": "https://github.com/CrayLabs/ml_lib_builder/releases/download/v0.2/libtensorflow-2.16-linux-x64-rocm-5.7.0.tgz"
-        },
         {
             "name": "onnxruntime",
             "version": "1.17.3",

diff --git a/smartsim/_core/_install/platform.py b/smartsim/_core/_install/platform.py
@@ -80,6 +80,7 @@ class Device(enum.Enum):
     CPU = "cpu"
     CUDA11 = "cuda-11"
     CUDA12 = "cuda-12"
+    ROCM5 = "rocm-5"
     ROCM6 = "rocm-6"
 
     @classmethod