[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
Re: PyTorch with ROCm
From: |
David Elsing |
Subject: |
Re: PyTorch with ROCm |
Date: |
Sun, 11 Aug 2024 19:11:41 +0000 |
Hello,
I updated PyTorch to 2.4.0 [1, 2] and the ROCm packages to 6.1.2.
This time, I opted to put the ROCm packages into a separate channel [3]
only for one version, because the packages in Guix-HPC are intended for
many ROCm versions and only include versions up to 5.7.1 at the moment.
The PyTorch package with ROCm is also in a separate channel [4].
I tried making the GPU architectures a Guile parameter, but this does
not work because it is evaluated when the package is lowered.
`with-parameters` [5] from `(guix gexp)` also does not work here,
because IIUC it is only intended for `%current-system` and
`%current-target-system` as indicated in `compile-parameterized`. Is
that correct?
Replacing the procedure `%amdgpu-targets` in
`guix-rocm/packages/rocm-base.scm` [3] with `module-define!` works
however, but is of course more of a hack.
Cheers,
David
[1] https://issues.guix.gnu.org/72471
[2] https://codeberg.org/dtelsing/Guix/src/branch/pytorch
[3] https://codeberg.org/dtelsing/Guix-ROCm
[4] https://codeberg.org/dtelsing/Guix-ROCm-ML
[5] https://guix.gnu.org/manual/en/html_node/G_002dExpressions.html
[Prev in Thread] |
Current Thread |
[Next in Thread] |
- Re: PyTorch with ROCm,
David Elsing <=