一、环境
显卡:Nvidia RTX 3090、3060、3080、3050、3070
系统:Ubuntu 16.04 18.04
CUDA:CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
CuDNN:NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
二、CUDA Toolkit安装
本教程无需预装新版本显卡驱动,在安装CUDA时会自动安装
首先进入CUDA Toolkit下载官网:CUDA Toolkit 11.7 Downloads | NVIDIA Developer
选择deb[local](选择其他的比如runfile等会报错),严格按照命令行逐行输入安装