python3Packages.tensorflow-bin: fix cuda routines that require jit (#449662)

2025-10-20 19:44:36 +00:00
parent eaa4d38662 98ad6ca117
commit ce268ef06c
1 changed files with 6 additions and 1 deletions
@@ -205,7 +205,12 @@ buildPythonPackage rec {
          ''}
        done
      ''
-    );
+    )
+    # Symlink nvcc besides TensorFlow so that routines that require JIT can work
+    # properly.
+    + lib.optionalString cudaSupport ''
+      ln -s ${cudaPackages.cuda_nvcc} "$out/${python.sitePackages}/tensorflow/cuda"
+    '';

  # Upstream has a pip hack that results in bin/tensorboard being in both tensorflow
  # and the propagated input tensorboard, which causes environment collisions.