fixing pytorch for cpu on linux

Ailurus1 · Dec 2, 2022 · e360247 · e360247
1 parent a622044
commit e360247
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py b/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py
@@ -5,6 +5,8 @@
 import os
 import shutil
 import numpy as np
+
+
 import torch
 import torchvision.models as models
 

diff --git a/cm-mlops/script/get-generic-python-lib/_cm.json b/cm-mlops/script/get-generic-python-lib/_cm.json
@@ -59,7 +59,6 @@
     "jax_cuda": {
       "env": {
         "CM_GENERIC_PYTHON_PACKAGE_NAME": "jax[cuda]",
-        "CM_GENERIC_PYTHON_PIP_EXTRA": "-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html",
         "CM_JAX_VERSION_EXTRA":"CUDA"
       },
       "new_env_keys": [

diff --git a/cm-mlops/script/install-generic-python-lib/_cm.json b/cm-mlops/script/install-generic-python-lib/_cm.json
@@ -119,7 +119,8 @@
     },
     "torch": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torch"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torch",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torch_cuda": {
@@ -138,7 +139,8 @@
     },
     "torchvision": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchvision"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchvision",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torchvision_cuda": {
@@ -157,7 +159,8 @@
     },
     "torchaudio": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchaudio"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchaudio",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torchaudio_cuda": {