fix Archlinux install error

1 files changed, 8 insertions, 0 deletions

diff --git a/install.sh b/install.sh
index bf6d29f..3f82874 100755
--- a/install.sh
+++ b/install.sh
@@ -51,6 +51,14 @@ if [[ `uname` == "Darwin" ]]; then
     export CC=clang
     export CXX=clang++
 fi
+# If we're on Arch linux, use cuda's gcc
+if [[ `uname -a` == *"ARCH"* ]]; then
+    export CC=/opt/cuda/bin/gcc
+    export CXX=/opt/cuda/bin/g++
+    if [[ -f /opt/cuda/bin/nvcc ]]; then {
+        path_to_nvcc=/opt/cuda/bin/nvcc
+    } fi
+fi
 
 echo "Installing Lua version: ${TORCH_LUA_VERSION}"
 mkdir -p install