diff options
author | David Elsing <david.elsing@posteo.net> | 2024-03-23 22:05:06 +0000 |
---|---|---|
committer | Ludovic Courtès <ludo@gnu.org> | 2024-06-18 11:45:07 +0200 |
commit | d218d024ede876e328da34bf4a71d4738e495c17 (patch) | |
tree | 81fb88ee812741e2f1f2160e48e97258a9247d8f /gnu/packages/machine-learning.scm | |
parent | 7b62d614e774031d3247be774e4e2ebe4d30fa39 (diff) | |
download | guix-d218d024ede876e328da34bf4a71d4738e495c17.tar guix-d218d024ede876e328da34bf4a71d4738e495c17.tar.gz |
gnu: Add python-pytorch-avx.
* gnu/packages/machine-learning.scm (python-pytorch-avx): New variable.
Diffstat (limited to 'gnu/packages/machine-learning.scm')
-rw-r--r-- | gnu/packages/machine-learning.scm | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index d80e8076de..83fb8d4dc1 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -4734,6 +4734,21 @@ PyTorch when needed. Note: currently this package does not provide GPU support.") (license license:bsd-3))) +;; This package variant includes the dependencies requiring at least AVX2 or +;; AVX-512. +(define-public python-pytorch-avx + (package/inherit python-pytorch + (name "python-pytorch-avx") + (inputs + (modify-inputs (package-inputs python-pytorch) + (append fbgemm nnpack))) + (arguments + (substitute-keyword-arguments (package-arguments python-pytorch) + ((#:phases phases) + #~(modify-phases #$phases + (delete 'disable-avx-dependencies))))) + (supported-systems '("x86_64-linux")))) + (define-public python-pytorch-for-r-torch (package (inherit python-pytorch) |