mirror/guix
1
0
Fork 0
mirror of https://git.savannah.gnu.org/git/guix.git synced 2025-01-20 06:37:08 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

gnu: Add python-pytorch-avx.

Browse source 
* gnu/packages/machine-learning.scm (python-pytorch-avx): New variable.
This commit is contained in:
David Elsing 2024-03-23 22:05:06 +00:00 committed by Ludovic Courtès
parent 7b62d614e7
commit d218d024ed
No known key found for this signature in database
GPG key ID: 090B11993D9AEBB5
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
gnu/packages/machine-learning.scm
View file

@ -4734,6 +4734,21 @@ (define-public python-pytorch
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)