Sven-Hendrik Haase pushed to branch main at Arch Linux / Packaging / Packages / 
python-pytorch


Commits:
4ba853b0 by Sven-Hendrik Haase at 2025-02-19T10:32:49+01:00
upgpkg: 2.6.0-5: Re-enable Nvidia Pascal architecture

Fixes #19.

- - - - -


2 changed files:

- .SRCINFO
- PKGBUILD


Changes:

=====================================
.SRCINFO
=====================================
@@ -1,7 +1,7 @@
 pkgbase = python-pytorch
        pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU 
acceleration
        pkgver = 2.6.0
-       pkgrel = 4
+       pkgrel = 5
        url = https://pytorch.org
        arch = x86_64
        license = BSD


=====================================
PKGBUILD
=====================================
@@ -8,7 +8,7 @@ pkgname=("${pkgbase}" "${pkgbase}-opt" "${pkgbase}-cuda" 
"${pkgbase}-opt-cuda" "
 # When updating pytorch, also check the compatibility table for torchvision
 # https://github.com/pytorch/vision?tab=readme-ov-file#installation
 pkgver=2.6.0
-pkgrel=4
+pkgrel=5
 _pkgdesc='Tensors and Dynamic neural networks in Python with strong GPU 
acceleration'
 pkgdesc="${_pkgdesc}"
 arch=('x86_64')
@@ -239,7 +239,7 @@ _prepare() {
   export TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
 
   # This list is from ./Dockerfile
-  export TORCH_CUDA_ARCH_LIST="7.0 7.2 7.5 8.0 8.6 8.7 8.9 9.0 9.0a"
+  export TORCH_CUDA_ARCH_LIST="6.1 7.0 7.2 7.5 8.0 8.6 8.7 8.9 9.0 9.0a"
 
   export ROCM_PATH=/opt/rocm
   export HIP_ROOT_DIR=/opt/rocm



View it on GitLab: 
https://gitlab.archlinux.org/archlinux/packaging/packages/python-pytorch/-/commit/4ba853b0197e5515b7e9356bc0b17ac07e9e5aef

-- 
View it on GitLab: 
https://gitlab.archlinux.org/archlinux/packaging/packages/python-pytorch/-/commit/4ba853b0197e5515b7e9356bc0b17ac07e9e5aef
You're receiving this email because of your account on gitlab.archlinux.org.


Reply via email to