Date: Wednesday, March 27, 2019 @ 17:56:04 Author: kgizdov Revision: 444538
build for CUDA with GCC8 Modified: python-pytorch/trunk/PKGBUILD ----------+ PKGBUILD | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2019-03-27 17:23:10 UTC (rev 444537) +++ PKGBUILD 2019-03-27 17:56:04 UTC (rev 444538) @@ -1,4 +1,5 @@ # Maintainer: Sven-Hendrik Haase <[email protected]> +# Co-Maintainer: Konstantin Gizdov <[email protected]> # Contributor: Stephen Zhang <zsrkmyn at gmail dot com> pkgbase="python-pytorch" @@ -5,13 +6,13 @@ pkgname=("python-pytorch" "python-pytorch-cuda") _pkgname="pytorch" pkgver=1.0.1 -pkgrel=3 +pkgrel=4 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://pytorch.org"; license=('BSD') depends=('python' 'python-yaml' 'python-numpy' 'opencv' 'nccl') -makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'cuda' 'cudnn' 'git' 'gcc7') +makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'cuda' 'cudnn' 'git' 'gcc') source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$pkgver"; "git+https://github.com/catchorg/Catch2"; "git+https://github.com/pybind/pybind11"; @@ -18,7 +19,6 @@ "git+https://github.com/NVlabs/cub"; "git+https://github.com/eigenteam/eigen-git-mirror"; "git+https://github.com/google/googletest"; - "git+https://github.com/NervanaSystems/nervanagpu"; "git+https://github.com/google/benchmark"; "git+https://github.com/google/protobuf"; "git+https://github.com/Yangqing/ios-cmake"; @@ -64,7 +64,6 @@ 'SKIP' 'SKIP' 'SKIP' - 'SKIP' 'SKIP') prepare() { @@ -76,7 +75,6 @@ git config submodule."third_party/cub".url "${srcdir}"/cub git config submodule."third_party/eigen".url "${srcdir}"/eigen-git-mirror git config submodule."third_party/googletest".url "${srcdir}"/googletest - git config submodule."third_party/nervanagpu".url "${srcdir}"/nervanagpu git config submodule."third_party/benchmark".url "${srcdir}"/benchmark git config submodule."third_party/protobuf".url "${srcdir}"/protobuf git config submodule."third_party/ios-cmake".url "${srcdir}"/ios-cmake @@ -107,8 +105,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 + export CC=gcc + export CXX=g++ export PYTORCH_BUILD_VERSION=${pkgver} export PYTORCH_BUILD_NUMBER=1 @@ -124,7 +122,7 @@ echo "Building with cuda" export NO_CUDA=0 export WITH_CUDNN=1 - export CUDAHOSTCXX=g++-7 + export CUDAHOSTCXX=g++ export CUDA_HOME=/opt/cuda export CUDNN_LIB_DIR=/usr/lib export CUDNN_INCLUDE_DIR=/usr/include
