fix GPU detection error

1 files changed, 2 insertions, 5 deletions

diff --git a/PKGBUILD b/PKGBUILD
index c42d504648d7..90a1550076e2 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 \