nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-27 01:13:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #232480 from wegank/cudatoolkit-eval-2

Browse Source 
python310Packages.torchWithCuda: add platform check
This commit is contained in:
Weijia Wang 2023-05-17 22:56:07 +03:00 committed by GitHub
commit ee8921a137
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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