[arch-commits] Commit in python-pytorch/trunk (PKGBUILD)
Sven-Hendrik Haase
svenstaro at archlinux.org
Sun Dec 9 17:23:44 UTC 2018
Date: Sunday, December 9, 2018 @ 17:23:43
Author: svenstaro
Revision: 413039
Prepare 1.0.0
Modified:
python-pytorch/trunk/PKGBUILD
----------+
PKGBUILD | 23 +++++++++--------------
1 file changed, 9 insertions(+), 14 deletions(-)
Modified: PKGBUILD
===================================================================
--- PKGBUILD 2018-12-09 17:20:52 UTC (rev 413038)
+++ PKGBUILD 2018-12-09 17:23:43 UTC (rev 413039)
@@ -4,8 +4,8 @@
pkgbase="python-pytorch"
pkgname=("python-pytorch" "python-pytorch-cuda")
_pkgname="pytorch"
-pkgver=1.0rc1
-pkgrel=6
+pkgver=1.0.0
+pkgrel=1
pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration"
arch=('x86_64')
url="https://pytorch.org"
@@ -38,9 +38,7 @@
"git+https://github.com/USCILab/cereal"
"git+https://github.com/onnx/onnx-tensorrt"
"git+https://github.com/shibatch/sleef"
- "git+https://github.com/intel/ideep"
- 12116.patch
- opencv4.patch)
+ "git+https://github.com/intel/ideep")
sha256sums=('SKIP'
'SKIP'
'SKIP'
@@ -67,16 +65,11 @@
'SKIP'
'SKIP'
'SKIP'
- 'SKIP'
- '2a2cd21dbdf7253185c8835a3f36b543a4b1655d837e01f7cfd27ab81819f2d5'
- '8cf4226099f92315f14c83066f77e44443bc3d35aedf94d99b910f035cc9bc90')
+ 'SKIP')
prepare() {
cd "${_pkgname}-${pkgver}"
- patch -Np1 -i "${srcdir}"/12116.patch
- patch -Np1 -i "${srcdir}"/opencv4.patch
-
git submodule init
git config submodule."third_party/catch".url "${srcdir}"/Catch2
git config submodule."third_party/pybind11".url "${srcdir}"/pybind11
@@ -114,6 +107,8 @@
build() {
# Uncomment and modify the following line to enable Intel MKL and magma support
# export CMAKE_PREFIX_PATH=/opt/intel/mkl/include:/opt/intel/mkl/lib/intel64:/opt/magma \
+ export CC=gcc-7
+ export CXX=g++-7
echo "Building without cuda"
export NO_CUDA=1
@@ -125,8 +120,6 @@
python setup.py build
echo "Building with cuda"
- export CC=gcc-7
- export CXX=g++-7
export NO_CUDA=0
export WITH_CUDNN=1
export CUDAHOSTCXX=g++-7
@@ -133,7 +126,9 @@
export CUDA_HOME=/opt/cuda
export CUDNN_LIB_DIR=/opt/cuda/lib64
export CUDNN_INCLUDE_DIR=/opt/cuda/include
- export TORCH_CUDA_ARCH_LIST="3.7;5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5"
+ export TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1 7.0+PTX"
+ 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"
cd "$srcdir/${_pkgname}-${pkgver}-cuda"
python setup.py build
More information about the arch-commits
mailing list