talexander/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Releases Activity

pkgsCuda: init

Browse Source 
Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
This commit is contained in:
Connor Baker 2025-06-10 20:07:15 +00:00
parent ef768dbd50
commit d43affbd43
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkgs/top-level/variants.nix
View File

@ -95,6 +95,15 @@ self: super: {
};
});
# Full package set with cuda on rocm off
# Mostly useful for asserting pkgs.pkgsCuda.torchWithCuda == pkgs.torchWithCuda and similar
pkgsCuda = nixpkgsFun {
config = super.config // {
cudaSupport = true;
rocmSupport = false;
};
};
# `pkgsForCudaArch` maps each CUDA capability in _cuda.db.cudaCapabilityToInfo to a Nixpkgs variant configured for
# that target system. For example, `pkgsForCudaArch.sm_90a.python3Packages.torch` refers to PyTorch built for the
# Hopper architecture, leveraging architecture-specific features.