nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-05 19:53:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.torchWithRocm: ignore config.cudaSupport

Browse Source
This commit is contained in:
Someone Serge 2023-03-21 16:30:06 +02:00
parent a9faf1b9ef
commit 0f76efb481
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/top-level/python-packages.nix
View File

@ -11856,6 +11856,7 @@ self: super: with self; {
torchWithCuda = self.torch.override {
magma = pkgs.magma-cuda;
cudaSupport = true;
rocmSupport = false;
};
torchWithoutCuda = self.torch.override {
@ -11865,6 +11866,7 @@ self: super: with self; {
torchWithRocm = self.torch.override {
magma = pkgs.magma-hip;
rocmSupport = true;
cudaSupport = false;
};
torchWithoutRocm = self.torch.override {