diff options
author | Maarten de Vries | 2020-11-27 11:15:59 +0100 |
---|---|---|
committer | Maarten de Vries | 2020-11-27 11:18:40 +0100 |
commit | 0a98b6cf11783f3160b82c23f442a6c7ab62e595 (patch) | |
tree | 6bcb789fe15ed435f0006de79f4397005bc4fa27 | |
parent | 6bc22c9966f349fa52b3bc0a16ef3921a86c0701 (diff) | |
download | aur-0a98b6cf11783f3160b82c23f442a6c7ab62e595.tar.gz |
Remove deprecated cuda architectures.
-rw-r--r-- | .SRCINFO | 2 | ||||
-rw-r--r-- | PKGBUILD | 8 |
2 files changed, 5 insertions, 5 deletions
@@ -1,7 +1,7 @@ pkgbase = torchvision-cuda pkgdesc = datasets, transforms, and models specific to computer vision (C++ library only) pkgver = 0.8.1 - pkgrel = 1 + pkgrel = 2 url = https://github.com/pytorch/vision arch = x86_64 license = BSD @@ -3,7 +3,7 @@ pkgname=torchvision-cuda pkgdesc='datasets, transforms, and models specific to computer vision (C++ library only)' pkgver=0.8.1 -pkgrel=1 +pkgrel=2 license=("BSD") url='https://github.com/pytorch/vision' arch=('x86_64') @@ -37,9 +37,9 @@ prepare() { -DCMAKE_BUILD_TYPE=Release \ -DCMAKE_CXX_FLAGS="${CXXFLAGS} -O3" \ -DWITH_CUDA=ON \ - -DTORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" \ - -DCUDA_ARCH_LIST="3.5;3.7;5.0;5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" \ - -DCMAKE_CUDA_ARCHITECTURES="35;37;50;52;53;60;60-virtual;61;61-virtual;62;62-virtual;70;70-virtual;72;72-virtual;75;75-virtual" + -DTORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" \ + -DCUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" \ + -DCMAKE_CUDA_ARCHITECTURES="52;53;60;60-virtual;61;61-virtual;62;62-virtual;70;70-virtual;72;72-virtual;75;75-virtual" } |