Ubuntu21.04安装cuda10.2和cudnn

系统:Ubuntu21.04

显卡:NVIDIA 1080Ti

cuda是NVIDIA公司设计用来加速GPU训练的,而cudnn则是可以把cuda应用到深度学习领域,如果想用GPU跑深度神经网络的话,两者缺一不可。

之前一直有一个知识误区,以为nvidia-smi输出界面中cuda version就是本机的cuda版本,在跑程序的时候发现根本找不到cuda,后面查阅资料才知道这里的cuda version指的是当前显卡驱动支持的最高cuda版本。自己电脑上真正的cuda版本需要通过nvcc -V才能查看,一查发现原来自己电脑上根本没装cuda哈哈,于是在电脑上安装了cuda10.2以及对应的cudnn

注意:如果你输入nvidia-smi啥反映都没,说明你的电脑显卡驱动都还没安装,需要先装一下显卡驱动。

安装CUDA顺序如下:

Ubuntu18.04安装CUDA10.2 + cuDNN v7.6.5_坚持健身的程序员的博客-CSDN博客_ubuntu安装cuda10.2

这篇文章写的过程很清晰,咱们21.04版本基本可以参照上面这篇博客来装cuda10.2

cuda安装好后,后面cudnn的安装建议参照下面这篇博客,因为官网现在下载好像都只有deb文件了:

Ubuntu18.04+CUDA10.2+cuDNN10.2_A_first的博客-CSDN博客_ubuntu18.04安装cuda10.2

你可能感兴趣的:(python,深度学习)