2.2.1　Nvidia 10/20/30/40系列显卡选择的GPU版本

由于40系显卡的推出，目前市场上会有Nvidia 10、20、30、40系列显卡并存的情况。对于需要调用专用编译器的PyTorch来说，不同的显卡需要安装不同的依赖计算包，作者在此总结了不同显卡的PyTorch版本以及CUDA和cuDNN的对应关系，如表2-1所示。

注意：这里的区别主要在于显卡运算库CUDA与cuDNN的区别，当在20/30/40系列显卡上使用PyTorch时，可以安装11.6以上版本以及cuDNN 8.1以上版本的计算包，而在10系列版本的显卡上，建议优先使用2.0版本以前的PyTorch。

下面以CUDA 11.7+cuDNN 8.2.0组合为例，演示完整的PyTorch 2.0 GPU Nvidia运行库的安装步骤，其他不同版本CUDA+cuDNN组合的安装过程基本一致。