summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorStephen Zhang2017-03-16 13:41:55 +0800
committerStephen Zhang2017-03-16 13:41:55 +0800
commited3a82e3af6f4d6d9a2d75338a5a91d050d27c69 (patch)
tree0c62cc66cca08862e4f7a8f036cb3a24e1b089fb
parent6585044547ad84765fb1eb9d71edfb7cc136db11 (diff)
downloadaur-ed3a82e3af6f4d6d9a2d75338a5a91d050d27c69.tar.gz
fix GPU detection error
-rw-r--r--PKGBUILD7
1 files changed, 2 insertions, 5 deletions
diff --git a/PKGBUILD b/PKGBUILD
index c42d504648d..90a1550076e 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -26,11 +26,6 @@ pkgver () {
prepare() {
cd "$srcdir/"
- # FindCUDA module may cause GPU detection failed
- sed -i -e '39s#-DCMAKE_MODULE_PATH="$BASE_DIR/cmake/FindCUDA"##g' \
- "${_pkgname}/torch/lib/build_all.sh"
- rm -r "${_pkgname}/cmake/FindCUDA/"
-
cp -a "${_pkgname}" "${_pkgname}-py2"
cd "${_pkgname}"
sed -e "s|#![ ]*/usr/bin/python$|#!/usr/bin/python2|" \
@@ -43,6 +38,7 @@ build() {
msg "Building Python 2"
cd "$srcdir/${_pkgname}-py2"
CC=gcc-5 \
+ CXX=g++-5 \
WITH_CUDA=1 \
CUDA_HOME=/opt/cuda \
WITH_CUDNN=1 \
@@ -53,6 +49,7 @@ build() {
msg "Building Python 3"
cd "$srcdir/${_pkgname}"
CC=gcc-5 \
+ CXX=g++-5 \
WITH_CUDA=1 \
CUDA_HOME=/opt/cuda \
WITH_CUDNN=1 \