1 files changed, 9 insertions, 6 deletions
diff --git a/abs/core/nvidia/nvidia.install b/abs/core/nvidia/nvidia.install
index 8e01d0a..c2c292a 100644
--- a/abs/core/nvidia/nvidia.install
+++ b/abs/core/nvidia/nvidia.install
@@ -1,15 +1,18 @@
 post_install() {
-    KERNEL_VERSION='2.6.39-LinHES'
-    depmod $KERNEL_VERSION
+    EXTRAMODULES='extramodules-3.6-ARCH'
+    depmod $(cat /usr/lib/modules/$EXTRAMODULES/version)
     echo 'In order to use nvidia module, reboot the system.'
 }
 
 post_upgrade() {
-    KERNEL_VERSION='2.6.39-LinHES'
-    depmod $KERNEL_VERSION
+    EXTRAMODULES='extramodules-3.6-ARCH'
+    depmod $(cat /usr/lib/modules/$EXTRAMODULES/version)
+    if  [ "$(vercmp $2 310.19-2)" -lt 0 ]; then
+        echo 'If your card is from the 7xxx series or earlier, install nvidia-304xx'
+    fi
 }
 
 post_remove() {
-    KERNEL_VERSION='2.6.39-LinHES'
-    depmod $KERNEL_VERSION
+    EXTRAMODULES='extramodules-3.6-ARCH'
+    depmod $(cat /usr/lib/modules/$EXTRAMODULES/version)
 }