Commit 504d7531 authored by Connor Baker's avatar Connor Baker
Browse files

torchvision: remove cudaSupport argument; must use whatever torch used

parent 55e0d9a9
Loading
Loading
Loading
Loading
+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
@@ -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!