Ubuntu16.04装CUDA和cuDNN
前言
前提是显卡驱动已经装好。可以参考Ubuntu16.04安装NVIDIA显卡驱动。
笔者要复现的代码环境是ubuntu16.04,Python2.7,后续还要装TensorFlow-GPU。
参考文献:
[1] 理清GPU、CUDA、CUDA Toolkit、cuDNN关系以及下载安装
[2] 显卡驱动与cuda对应关系,cudnn与cuda之间的关系
[3] Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
[4] CUDA Tookit Notes
[5] Ubuntu18.04+RTX 2080Ti+CUDA 10.0 +cuDNN+PyTorch搭建深度学习环境
终端运行nvidia-smi
,看图得知显卡的计算能力为460.67,最高可装11.2版本的CUDA。
因此,根据TensorFlow官网和自己的需求,选择Ubuntu16.04下装CUDA10.1,cuDNN7.6。
一、CUDA安装
1.查看ubuntu系统位数
终端输入以下命令获取ubuntu系统位数信息待用。
sudo uname --m