diff options
author | Daniël de Kok <me@danieldk.eu> | 2020-09-30 13:56:05 +0200 |
---|---|---|
committer | Daniël de Kok <me@danieldk.eu> | 2020-09-30 13:56:05 +0200 |
commit | 7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08 (patch) | |
tree | f8cd566552c6fb33c864d7e6b5b41db6feddd5d3 /pkgs/development/python-modules/tensorflow/2/default.nix | |
parent | b751c120cef6d42ed26093df5363800584c97c1f (diff) | |
download | nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar.gz nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar.bz2 nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar.lz nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar.xz nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.tar.zst nixpkgs-7fb2cc17251d9bc8c0879ba98dd2dd91f24c2b08.zip |
python3Packages.tensorflow_2: update CUDA capabilities
Sync CUDA capabilities with upstream: https://github.com/tensorflow/tensorflow/blob/b0c3b5bdd768a1410a498642aec6de842692d857/.bazelrc#L584
Diffstat (limited to 'pkgs/development/python-modules/tensorflow/2/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tensorflow/2/default.nix | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/tensorflow/2/default.nix b/pkgs/development/python-modules/tensorflow/2/default.nix index bd049a9a2ca..7925cb15f03 100644 --- a/pkgs/development/python-modules/tensorflow/2/default.nix +++ b/pkgs/development/python-modules/tensorflow/2/default.nix @@ -22,7 +22,7 @@ # XLA without CUDA is broken , xlaSupport ? cudaSupport # Default from ./configure script -, cudaCapabilities ? [ "3.5" "5.2" ] +, cudaCapabilities ? [ "sm_35" "sm_50" "sm_60" "sm_70" "sm_75" "compute_80" ] , sse42Support ? stdenv.hostPlatform.sse4_2Support , avx2Support ? stdenv.hostPlatform.avx2Support , fmaSupport ? stdenv.hostPlatform.fmaSupport |