mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-10-15 17:07:25 +03:00
torchvision: remove cudaSupport argument; must use whatever torch used
This commit is contained in:
parent
55e0d9a93d
commit
504d7531ba
@ -1,5 +1,4 @@
|
||||
{ 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!
|
||||
|
Loading…
Reference in New Issue
Block a user