Skip to content

Instantly share code, notes, and snippets.

@jonbakerfish
Last active August 27, 2017 09:32
Show Gist options
  • Save jonbakerfish/3f7178da72a26a6a7e0f53c5a5bf164a to your computer and use it in GitHub Desktop.
Save jonbakerfish/3f7178da72a26a6a7e0f53c5a5bf164a to your computer and use it in GitHub Desktop.
nvidia_node.sh
#!/bin/bash
/sbin/modprobe nvidia
if [ "$?" -eq 0 ]; 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
mknod -m 666 /dev/nvidia$i c 195 $i
done
mknod -m 666 /dev/nvidiactl c 195 255
else
exit 1
fi
/sbin/modprobe nvidia-uvm
if [ "$?" -eq 0 ]; then
# Find out the major device number used by the nvidia-uvm driver
D=`grep nvidia-uvm /proc/devices | awk '{print $1}'`
mknod -m 666 /dev/nvidia-uvm c $D 0
else
exit 1
fi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment