diff options
author | Stephen Zhang | 2017-03-16 05:49:02 +0800 |
---|---|---|
committer | Stephen Zhang | 2017-03-16 05:49:02 +0800 |
commit | 6585044547ad84765fb1eb9d71edfb7cc136db11 (patch) | |
tree | c64cff2d322984650f11a29b110e8876cdb4e8f9 | |
parent | bdf7526bbe985454c03997d750e8c00234dfa85d (diff) | |
download | aur-6585044547ad84765fb1eb9d71edfb7cc136db11.tar.gz |
fix GPU detection failed bug
-rw-r--r-- | .SRCINFO | 2 | ||||
-rw-r--r-- | PKGBUILD | 6 |
2 files changed, 7 insertions, 1 deletions
@@ -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 @@ -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|" \ |