From 2f75804f8afaa3f7398fa33beb6ec530f311140b Mon Sep 17 00:00:00 2001 From: Gaurav Saini <147703805+gauravsaini04@users.noreply.github.com> Date: Tue, 20 Aug 2024 03:03:17 +0530 Subject: [PATCH] [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 --- .../local-features/machine-learning-packages/install.sh | 2 +- 1 file changed, 1 insertion(+), 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 index 917a04d28..6f22a4cf2 100644 --- 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