fix pytorch cuda pin

aws · Nov 13, 2024 · a25cb70 · a25cb70
1 parent 506d90e
commit a25cb70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build_artifacts/v2/v2.1/v2.1.1/gpu.env.in b/build_artifacts/v2/v2.1/v2.1.1/gpu.env.in
@@ -12,7 +12,7 @@ conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.3,<0.2.0']
 conda-forge::langchain[version='>=0.2.16,<0.3.0']
 conda-forge::fastapi[version='>=0.115.2,<0.116.0']
 conda-forge::uvicorn[version='>=0.32.0,<0.33.0']
-conda-forge::pytorch[version='>=2.3.1,<2.4.0=*cuda12*']
+conda-forge::pytorch[version='>=2.3.1,<2.4.0',build='*cuda12*']
 conda-forge::tensorflow[version='>=2.17.0,<2.18.0']
 conda-forge::python[version='>=3.11.10,<3.12.0']
 conda-forge::pip[version='>=24.2,<24.3.0']