diff options
author | Matthew Bauer <mjbauer95@gmail.com> | 2020-08-19 13:34:18 -0500 |
---|---|---|
committer | Matthew Bauer <mjbauer95@gmail.com> | 2020-09-10 23:36:23 -0500 |
commit | 59eecacd2d2c0ebf32e393a2c111b38755004f53 (patch) | |
tree | 077d26b6a3dc9fcccbe0944239cc71a390794006 /pkgs/development/python-modules/tensorflow/2/default.nix | |
parent | 9cd240c14a573e8a1e74cc0d06d925920e6c6105 (diff) | |
download | nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar.gz nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar.bz2 nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar.lz nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar.xz nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.tar.zst nixpkgs-59eecacd2d2c0ebf32e393a2c111b38755004f53.zip |
python3Packages.tensorflow: use cudatoolkit 11
Diffstat (limited to 'pkgs/development/python-modules/tensorflow/2/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tensorflow/2/default.nix | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow/2/default.nix b/pkgs/development/python-modules/tensorflow/2/default.nix index f99a2590c44..bd049a9a2ca 100644 --- a/pkgs/development/python-modules/tensorflow/2/default.nix +++ b/pkgs/development/python-modules/tensorflow/2/default.nix @@ -46,6 +46,7 @@ let paths = [ cudatoolkit.lib cudatoolkit.out + ] ++ lib.optionals (lib.versionOlder cudatoolkit.version "11") [ # for some reason some of the required libs are in the targets/x86_64-linux # directory; not sure why but this works around it "${cudatoolkit}/targets/${stdenv.system}" |