public
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.tensorflow_2: update CUDA capabilities 

Sync CUDA capabilities with upstream:

b0c3b5bdd7/.bazelrc (L584)
This commit is contained in:
Daniël de Kok 2020-09-30 13:56:05 +02:00
parent b751c120ce
commit 7fb2cc1725
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/tensorflow/2/default.nix
View File

@ -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