Skip to content

Commit 9ede5ea

Browse files
[universal] - reduce size - install prev. rel. of torch (#1147)
1 parent 6fd137e commit 9ede5ea

File tree

1 file changed

+1
-1
lines changed
  • src/universal/.devcontainer/local-features/machine-learning-packages

1 file changed

+1
-1
lines changed

src/universal/.devcontainer/local-features/machine-learning-packages/install.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ if [[ "$(python --version)" != "" ]] && [[ "$(pip --version)" != "" ]]; then
4444
install_python_package "matplotlib"
4545
install_python_package "seaborn"
4646
install_python_package "scikit-learn"
47-
install_python_package "torch" "-f https://download.pytorch.org/whl/cpu/torch_stable.html"
47+
install_python_package "torch==2.3.1" "-f https://download.pytorch.org/whl/cpu/torch_stable.html"
4848
install_python_package "requests"
4949
install_python_package "plotly"
5050
else

0 commit comments

Comments
 (0)