python3Packages.tensorflow: use cudatoolkit 11
This commit is contained in:
parent
9cd240c14a
commit
59eecacd2d
|
@ -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}"
|
||||
|
|
|
@ -6860,9 +6860,9 @@ in {
|
|||
|
||||
tensorflow-build_2 = callPackage ../development/python-modules/tensorflow/2 {
|
||||
cudaSupport = pkgs.config.cudaSupport or false;
|
||||
cudatoolkit = pkgs.cudatoolkit_10;
|
||||
cudnn = pkgs.cudnn_cudatoolkit_10;
|
||||
nccl = pkgs.nccl_cudatoolkit_10;
|
||||
cudatoolkit = pkgs.cudatoolkit_11;
|
||||
cudnn = pkgs.cudnn_cudatoolkit_11;
|
||||
nccl = pkgs.nccl_cudatoolkit_11;
|
||||
openssl = pkgs.openssl_1_1;
|
||||
inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security;
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue