remove pytorch-gpu but restrict pytorch version

aws · Oct 17, 2024 · e5c9c9d · e5c9c9d
1 parent 636cb2e
commit e5c9c9d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/build_artifacts/v2/v2.1/v2.1.0/gpu.env.in b/build_artifacts/v2/v2.1/v2.1.0/gpu.env.in
@@ -12,8 +12,7 @@ conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.3,<1.0.0']
 conda-forge::langchain[version='>=0.2.5,<1.0.0']
 conda-forge::fastapi[version='>=0.110.3,<1.0.0']
 conda-forge::uvicorn[version='>=0.30.6,<1.0.0']
-conda-forge::pytorch[version='>=2.3.1,<3.0.0']
-conda-forge::pytorch-gpu[version='>=2.3.1,<2.4.1']
+conda-forge::pytorch[version='>=2.3.1,<2.4.0']
 conda-forge::tensorflow[version='>=2.17.0,<3.0.0']
 conda-forge::python[version='>=3.11.9,<3.12.0']
 conda-forge::pip[version='>=24.2,<25.0.0']