Commit 4aa67c95 authored by hacker1024's avatar hacker1024
Browse files

python311Packages.tensorflow-bin: Use CUDA 12

parent a5fcc930

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

+2 −3
Original line number Diff line number Diff line
@@ -140,12 +140,11 @@ buildPythonPackage rec { 
  postFixup =

    let

      # rpaths we only need to add if CUDA is enabled.

      cudapaths = lib.optionals cudaSupport ([

      cudapaths = lib.optionals cudaSupport [

        cudatoolkit.out

        cudatoolkit.lib

      ] ++ lib.optionals (!cudaPackages.cudaFlags.isJetsonBuild) [

        cudnn

      ]);

      ];



      libpaths = [

        stdenv.cc.cc.lib

pkgs/top-level/python-packages.nix

+1 −1
Original line number Diff line number Diff line
@@ -15392,7 +15392,7 @@ self: super: with self; { 
  tensorflow-bin = callPackage ../development/python-modules/tensorflow/bin.nix {

    inherit (pkgs.config) cudaSupport;

    # https://www.tensorflow.org/install/source#gpu

    cudaPackages = pkgs.cudaPackages_11;

    cudaPackages = pkgs.cudaPackages_12;

  };















  tensorflow-build = let