/etc/local.d/cuda.stop

cuda.stop (text/plain), 526 bytes, created by Tomas Vondra on 2011-07-26 21:00:56 UTC

#!/bin/bash

# check if there's a nvidia module
mod=`lsmod | grep '^nvidia ' | wc -l`

if [ "$mod" -eq "1" ]; then

	# Count the number of NVIDIA controllers found.
	NVDEVS=`lspci | grep -i NVIDIA`
	N3D=`echo "$NVDEVS" | grep "3D controller" | wc -l`
	NVGA=`echo "$NVDEVS" | grep "VGA compatible controller" | wc -l`
	N=`expr $N3D + $NVGA - 1`

	for i in `seq 0 $N`; do
		if [ -e "/dev/nvidia$i" ]; then
			rm /dev/nvidia$i
		fi
	done

	if [ -e "/dev/nvidiactl" ]; then
		rm /dev/nvidiactl
	fi

	rmmod nvidia

else
	exit 1
fi