From 11d3abe84cc5a7c752d93e1aea8c7571ea70182d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eric=20M=C3=BCller?= <mueller@kip.uni-heidelberg.de>
Date: Mon, 17 Mar 2025 11:54:02 +0100
Subject: [PATCH] fix(py-tensorflow): missing conf env vars for @2.18:
---
packages/py-tensorflow/package.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/packages/py-tensorflow/package.py b/packages/py-tensorflow/package.py
index 436a1ea5..6b01431b 100644
--- a/packages/py-tensorflow/package.py
+++ b/packages/py-tensorflow/package.py
@@ -728,6 +728,18 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension):
# Please specify the cuDNN version you want to use
env.set("TF_CUDNN_VERSION", spec["cudnn"].version.up_to(1))
+ # Please specify the hermetic CUDA version you want to use
+ env.set("HERMETIC_CUDA_VERSION", spec["cuda"].version.up_to(3))
+
+ # Please specify the hermetic cuDNN version you want to use
+ env.set("HERMETIC_CUDNN_VERSION", spec["cudnn"].version.up_to(3))
+
+ # Please specify the local CUDA path you want to use
+ env.set("LOCAL_CUDA_PATH", spec["cuda"].prefix)
+
+ # Please specify the local CUDNN path you want to use
+ env.set("LOCAL_CUDNN_PATH", spec["cudnn"].prefix)
+
if "+nccl" in spec:
cuda_paths.append(spec["nccl"].prefix)
@@ -737,6 +749,9 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension):
# Please specify the location where NCCL is installed
env.set("NCCL_INSTALL_PATH", spec["nccl"].prefix)
env.set("NCCL_HDR_PATH", spec["nccl"].prefix.include)
+
+ # Please specify the local NCCL path you want to use
+ env.set("LOCAL_NCCL_PATH", spec["nccl"].prefix)
else:
env.unset("TF_NCCL_VERSION")
--
GitLab