Loading pkgs/development/python-modules/torchvision/default.nix +1 −2 Original line number Diff line number Diff line { buildPythonPackage , cudaSupport ? torch.cudaSupport or false # by default uses the value from torch , fetchFromGitHub , lib , libjpeg_turbo Loading @@ -15,7 +14,7 @@ }: let inherit (torch) cudaPackages cudaCapabilities; inherit (torch) cudaCapabilities cudaPackages cudaSupport; inherit (cudaPackages) cudatoolkit cudaFlags cudaVersion; # NOTE: torchvision doesn't use cudnn; torch does! Loading Loading
pkgs/development/python-modules/torchvision/default.nix +1 −2 Original line number Diff line number Diff line { buildPythonPackage , cudaSupport ? torch.cudaSupport or false # by default uses the value from torch , fetchFromGitHub , lib , libjpeg_turbo Loading @@ -15,7 +14,7 @@ }: let inherit (torch) cudaPackages cudaCapabilities; inherit (torch) cudaCapabilities cudaPackages cudaSupport; inherit (cudaPackages) cudatoolkit cudaFlags cudaVersion; # NOTE: torchvision doesn't use cudnn; torch does! Loading