ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-09-30 09:37:59 +03:00
Code Issues Projects Releases Wiki Activity

python310Packages.torchWithCuda: add platform check

Browse Source
This commit is contained in:
Weijia Wang 2023-05-17 22:01:44 +03:00
parent d1bc3181fd
commit 9775ce1bd3
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkgs/development/python-modules/torch/default.nix
View File

@ -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