nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-11 06:34:13 +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 { torchWithCuda = self.torch.override {
magma = pkgs.magma-cuda; magma = pkgs.magma-cuda;
cudaSupport = true; cudaSupport = true;
rocmSupport = false;
}; };
torchWithoutCuda = self.torch.override { torchWithoutCuda = self.torch.override {
@ -11865,6 +11866,7 @@ self: super: with self; {
torchWithRocm = self.torch.override { torchWithRocm = self.torch.override {
magma = pkgs.magma-hip; magma = pkgs.magma-hip;
rocmSupport = true; rocmSupport = true;
cudaSupport = false;
}; };
torchWithoutRocm = self.torch.override { torchWithoutRocm = self.torch.override {