mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-09-30 01:17:28 +03:00
Merge pull request #232480 from wegank/cudatoolkit-eval-2
python310Packages.torchWithCuda: add platform check
This commit is contained in:
commit
ee8921a137
@ -55,6 +55,7 @@ let
|
||||
inherit (cudaPackages) cudatoolkit cudaFlags cudnn nccl;
|
||||
in
|
||||
|
||||
assert cudaSupport -> stdenv.isLinux;
|
||||
assert cudaSupport -> (cudaPackages.cudaMajorVersion == "11");
|
||||
|
||||
# confirm that cudatoolkits are sync'd across dependencies
|
||||
|
Loading…
Reference in New Issue
Block a user