Load cuda at runtime. Include cuda header files inside the project

1 files changed, 4 insertions, 4 deletions

diff --git a/install_ubuntu.sh b/install_ubuntu.sh
index 251da8c..0f1388c 100755
--- a/install_ubuntu.sh
+++ b/install_ubuntu.sh
@@ -5,19 +5,19 @@ cd "$script_dir"
 
 [ $(id -u) -ne 0 ] && echo "You need root privileges to run the install script" && exit 1
 
-dpkg -l nvidia-cuda-dev > /dev/null 2>&1
+dpkg -l cuda > /dev/null 2>&1
 cuda_missing="$?"
 
 set -e
-apt-get -y install build-essential nvidia-cuda-dev\
+apt-get -y install build-essential cuda\
 	libswresample-dev libavformat-dev libavcodec-dev libavutil-dev\
 	libx11-dev libxcomposite-dev\
 	libglew-dev libglfw3-dev\
 	libpulse-dev
 
 dependencies="glew libavcodec libavformat libavutil x11 xcomposite glfw3 libpulse libswresample"
-includes="$(pkg-config --cflags $dependencies) -I/opt/cuda/targets/x86_64-linux/include"
-libs="$(pkg-config --libs $dependencies) /usr/lib/x86_64-linux-gnu/stubs/libcuda.so -ldl -pthread -lm"
+includes="$(pkg-config --cflags $dependencies) -Iinclude"
+libs="$(pkg-config --libs $dependencies) -ldl -pthread -lm"
 g++ -c src/sound.cpp -O2 $includes
 g++ -c src/main.cpp -O2 $includes
 g++ -o gpu-screen-recorder -O2 sound.o main.o -s $libs