-
查看显卡型号
lspci | grep -i nvidia
-
查看合适的驱动版本
ubuntu-drivers devices 我的推荐版本是recommend : nvidia-driver-470
-
装推荐版本的驱动
sudo ubuntu-drivers autoinstall
-
查看系统内核版本
uname -r
-
默认驱动安装位置
/usr/src/nvidia-470.86
-
nvidia-smi
查看驱动版本,和最高支持CUDA版本(不是装的版本)
如果显示NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver.
大概率是内核更新了,在开机首选项那里改一个低版本的内核。 -
window键,搜nvidia,可以查看驱动版本信息
-
装CUDA
在官网下载适合(GPU型号,显卡驱动,神经网络支持)版本的CUDA,我下载的是cuda-8.0
sudo sh cuda-8.0.61.375.26-linux.run -override
除了拒绝装驱动,其他都选择默认,默认安装位置:/usr/local/cuda-8.0
,/usr/local/cuda
自动链接到它 -
cuDNN , GPU加速库
我装的V5(5.27.2016) for cuda 8.0 -
caffe-segnet-cudnn5
sudo make all sudo make pycaffe sudo test sudo make runtest 没过 unknow layertype HDFData Bias Dum...
05-20
1695
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)