Date: Wednesday, September 30, 2020 @ 07:41:33 Author: svenstaro Revision: 714060
upgpkg: python-pytorch 1.6.0-5: cuda 11.1 rebuild Modified: python-pytorch/trunk/PKGBUILD ----------+ PKGBUILD | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2020-09-30 07:13:29 UTC (rev 714059) +++ PKGBUILD 2020-09-30 07:41:33 UTC (rev 714060) @@ -6,7 +6,7 @@ _pkgname="pytorch" pkgver=1.6.0 _pkgver=1.6.0 -pkgrel=4 +pkgrel=5 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://pytorch.org" @@ -74,12 +74,12 @@ # export USE_SYSTEM_LIBS=ON export NCCL_VERSION=$(pkg-config nccl --modversion) export NCCL_VER_CODE=$(sed -n 's/^#define NCCL_VERSION_CODE\s*\(.*\).*/\1/p' /usr/include/nccl.h) - export CUDAHOSTCXX=g++-9 + export CUDAHOSTCXX=g++ export CUDA_HOME=/opt/cuda export CUDNN_LIB_DIR=/usr/lib export CUDNN_INCLUDE_DIR=/usr/include export TORCH_NVCC_FLAGS="-Xfatbin -compress-all" - export TORCH_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;8.0;8.0+PTX;" + export TORCH_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;8.0" # Not yet supported in pytorch: 8.0+PTX;8.6;8.6+PTX } build() {