[arch-commits] Commit in python-pytorch/trunk (PKGBUILD)
Sven-Hendrik Haase
svenstaro at archlinux.org
Wed Sep 4 12:08:52 UTC 2019
Date: Wednesday, September 4, 2019 @ 12:08:51
Author: svenstaro
Revision: 508012
Unfuck CUDA
Modified:
python-pytorch/trunk/PKGBUILD
----------+
PKGBUILD | 10 ++++------
1 file changed, 4 insertions(+), 6 deletions(-)
Modified: PKGBUILD
===================================================================
--- PKGBUILD 2019-09-04 11:56:05 UTC (rev 508011)
+++ PKGBUILD 2019-09-04 12:08:51 UTC (rev 508012)
@@ -61,13 +61,11 @@
export CUDNN_INCLUDE_DIR=/usr/include
export TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
export TORCH_CUDA_ARCH_LIST="3.0;3.2;3.5;3.7;5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5"
- # Experimental
- export _GLIBCXX_USE_CXX11_ABI=TRUE
}
build() {
echo "Building without cuda and without non-x86-64 optimizations"
- export USE_CUDA=1
+ export USE_CUDA=0
export USE_CUDNN=0
cd "${srcdir}/${_pkgname}-${pkgver}"
python setup.py build
@@ -74,7 +72,7 @@
echo "Building without cuda and with non-x86-64 optimizations"
- export USE_CUDA=1
+ export USE_CUDA=0
export USE_CUDNN=0
cd "${srcdir}/${_pkgname}-${pkgver}-opt"
echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
@@ -82,7 +80,7 @@
echo "Building with cuda and without non-x86-64 optimizations"
- export USE_CUDA=0
+ export USE_CUDA=1
export USE_CUDNN=1
cd "${srcdir}/${_pkgname}-${pkgver}-cuda"
python setup.py build
@@ -89,7 +87,7 @@
echo "Building with cuda and with non-x86-64 optimizations"
- export USE_CUDA=0
+ export USE_CUDA=1
export USE_CUDNN=1
cd "${srcdir}/${_pkgname}-${pkgver}-opt-cuda"
echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
More information about the arch-commits
mailing list