update gpu.env.in

aws · Jan 8, 2025 · 102615f · 102615f
1 parent ae8117b
commit 102615f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/build_artifacts/v1/v1.12/v1.12.0/gpu.env.in b/build_artifacts/v1/v1.12/v1.12.0/gpu.env.in
@@ -9,8 +9,8 @@ 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-gpu[version='>=2.0.0,<3.0.0']
-conda-forge::tensorflow-gpu[version='>=2.14.0,<3.0.0']
+conda-forge::pytorch[version='>=2.0.0,<3.0.0',build='*cuda11*']
+conda-forge::tensorflow[version='>=2.14.0,<3.0.0',build='*cuda11*']
 conda-forge::python[version='>=3.10.14,<3.11.0']
 conda-forge::pip[version='>=23.3.2,<24.0.0']
 conda-forge::torchvision[version='>=0.15.2,<1.0.0']
@@ -54,3 +54,4 @@ conda-forge::pyhive[version='>=0.7.0,<1.0.0']
 conda-forge::python-gssapi[version='>=1.8.3,<2.0.0']
 conda-forge::libmamba[version='<2.0.0']
 conda-forge::jupyter[version='1.1.1',build='pyhd8ed1ab_0']
+conda-forge::pyarrow-hotfix[version='>=0.6']