2 files changed, 7 insertions, 1 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 9191010d59e3..15cef2f74112 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,5 +1,5 @@
 # Generated by mksrcinfo v8
-# Wed Mar 15 19:48:33 UTC 2017
+# Wed Mar 15 21:48:52 UTC 2017
 pkgbase = python-pytorch-git
 	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
 	pkgver = 0.1.10.r46.gbe6322e4
diff --git a/PKGBUILD b/PKGBUILD
index 030a9eeee981..c42d504648d7 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -25,6 +25,12 @@ 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|" \