Commit cc47f003 authored by hacker1024's avatar hacker1024
Browse files

python3Packages.tensorflow-bin: Use cuda_compat on Jetsons

parent 90b1b8d6

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

+2 −1
Original line number Diff line number Diff line
@@ -95,7 +95,8 @@ buildPythonPackage { 
  build-system = [

    distutils

    wheel

  ] ++ lib.optionals cudaSupport [ addDriverRunpath ];

  ] ++ lib.optionals cudaSupport ([ addDriverRunpath ]

    ++ lib.optionals cudaPackages.cudaFlags.isJetsonBuild [ cudaPackages.autoAddCudaCompatRunpath ]);



  preConfigure = ''

    unset SOURCE_DATE_EPOCH