Unverified Commit ce268ef0 authored by Winter's avatar Winter Committed by GitHub
Browse files

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

parents eaa4d386 98ad6ca1

pkgs/development/python-modules/tensorflow/bin.nix

+6 −1
Original line number Diff line number Diff line
@@ -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.