Skip to content

1 pytorch 官网

pytorch Official website

2 pytorch 简介

2.1 认识pytorch

2.2 pytorch 软件栈

figure3

3 pytorch install

python
nvidia-smi

4 nvidia 相关软件库

4.1 显卡驱动

        显卡驱动的作用就是用来驱动显卡的,这是电脑硬件中所对应的一个软件。通过添加驱动程序计算机中的硬件就能正常的工作,当然不同的硬件使用的驱动程序也不一样。显卡对应的就是显卡驱动。

4.2 cuda

        统一计算设备架构(Compute Unified Device Architecture, CUDA),是由NVIDIA推出的通用并行计算架构。解决的是用更加廉价的设备资源,实现更高效的并行计算。 和中央处理器(Central Processing Unit, CPU)相对,图形处理器(Graphics Processing Unit, GPU)是显卡的核心芯片。而cuda正是英伟达开发的GPU的编程接口!

4.3 cudnn

        cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。.如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。总结来说,CPU适合串行计算,擅长逻辑控制。GPU擅长并行高强度并行计算,适用于AI算法的训练学习!

5 GPU

5.1 GPU 加速原理

H100 SM

5.2 最先进的GPU