driver、cuda、cudnn(GPU安装)
本文介绍一种GPU服务器安装深度学习框架,安装过程总共分为3步:
- 安装显卡驱动
- 安装cuda框架
- 安装cudnn框架
显卡驱动安装
添加源,
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
检查可安装的驱动版本,
ubuntu-drivers devices
选择合适的版本进行安装,
sudo apt install nvidia-driver-XXX
最后reboot
重启,然后输入nvidia-smi
即可查看驱动信息。
安装CUDA
CUDA各个版本的下载网站:https://developer.nvidia.com/cuda-toolkit-archive
选择==runfile(local)==进行安装,
将runfile添加可执行权限并执行,
chmod +x xxx.run
./xxx.run
在选择是否安装nvidia driver时,选择否,因为第一步已经安装过驱动了
然后修改环境变量,
sudo vim ~/.bashrc
在文件末尾加入,
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
最后,使配置生效,
source ~/.bashrc
cuda安装完毕。
安装CUDNN
下载cudnn,https://developer.nvidia.com/rdp/cudnn-download
执行安装,
sudo dpkg -i xxx.deb
CUDNN安装完毕。