1. 程式人生 > >Ubuntu 16.04 上安裝 CUDA 9.0 詳細教程

Ubuntu 16.04 上安裝 CUDA 9.0 詳細教程

#!/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


下次重啟時,你應該能直接看到/dev目錄下的三個nvidia的檔案
輸入:$ ls /dev/nvidia*
結果顯示:/dev/nvidia0       /dev/nvidiactl       /dev/nvidia-uvm