[Universal] - install latest cpu torch (#1164)

* [Universal] - install latest cpu pytorch * alternative method of torch installation without gpu support * change in method.. * Update src/universal/.devcontainer/local-features/machine-learning-packages/install.sh --------- Co-authored-by: Samruddhi Khandale <[email protected]>
devcontainers · Aug 19, 2024 · 2f75804 · 2f75804
1 parent 6b46a28
commit 2f75804
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh b/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh
@@ -44,7 +44,7 @@ if [[ "$(python --version)" != "" ]] && [[ "$(pip --version)" != "" ]]; then
     install_python_package "matplotlib"
     install_python_package "seaborn"
     install_python_package "scikit-learn"
-    install_python_package "torch==2.3.1" "-f https://download.pytorch.org/whl/cpu/torch_stable.html"
+    install_python_package "torch" "--index-url https://download.pytorch.org/whl/cpu"
     install_python_package "requests"
     install_python_package "plotly"
 else