summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorStephen Zhang2017-03-16 05:49:02 +0800
committerStephen Zhang2017-03-16 05:49:02 +0800
commit6585044547ad84765fb1eb9d71edfb7cc136db11 (patch)
treec64cff2d322984650f11a29b110e8876cdb4e8f9
parentbdf7526bbe985454c03997d750e8c00234dfa85d (diff)
downloadaur-6585044547ad84765fb1eb9d71edfb7cc136db11.tar.gz
fix GPU detection failed bug
-rw-r--r--.SRCINFO2
-rw-r--r--PKGBUILD6
2 files changed, 7 insertions, 1 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 9191010d59e..15cef2f7411 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 030a9eeee98..c42d504648d 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|" \