From 56e1a0d973a93715a12fd407f9e68e3c793ea681 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andrzej=20W=C3=B3jtowicz?= <andre@amu.edu.pl>
Date: Tue, 3 May 2016 19:34:25 +0200
Subject: [PATCH] updated cuBLAS libs versions, added linux headers
 installation and fixed typo

---
 install-env.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/install-env.sh b/install-env.sh
index 69d521d..fb7023d 100644
--- a/install-env.sh
+++ b/install-env.sh
@@ -374,7 +374,7 @@ function cublas_install {
 
     modprobe -r nouveau
 
-    apt-get install nvidia-drivers nvidia-modprobe libnvblas6.0 -y
+    apt-get install linux-headers-$(uname -r|sed 's,[^-]*-[^-]*-,,') nvidia-driver nvidia-modprobe libnvblas6.5 -y
 
     nvidia-modprobe
 
@@ -393,7 +393,7 @@ function cublas_check {
 
     echo "Started checking cuBLAS"
 
-    NVBLAS_CONFIG_FILE=/opt/blap-lib/cublas/nvblas.conf LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libnvblas.so.6.0 /usr/lib/x86_64-linux-gnu/libcublas.so.6.0" ${R_SAMPLE_BENCHMARK}
+    NVBLAS_CONFIG_FILE=/opt/blap-lib/cublas/nvblas.conf LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libnvblas.so.6.5 /usr/lib/x86_64-linux-gnu/libcublas.so.6.5" ${R_SAMPLE_BENCHMARK}
     
     echo "Finished checking cuBLAS"
 }