2.2.1　Nvidia 10/20/30/40系列顯卡選擇的GPU版本

由于40系顯卡的推出，目前市場(chǎng)上會(huì)有Nvidia 10、20、30、40系列顯卡并存的情況。對(duì)于需要調(diào)用專用編譯器的PyTorch來(lái)說(shuō)，不同的顯卡需要安裝不同的依賴計(jì)算包，作者在此總結(jié)了不同顯卡的PyTorch版本以及CUDA和cuDNN的對(duì)應(yīng)關(guān)系，如表2-1所示。

表2-1　 10/20/30/40系列顯卡的版本對(duì)比

注意：這里的區(qū)別主要在于顯卡運(yùn)算庫(kù)CUDA與cuDNN的區(qū)別，當(dāng)在20/30/40系列顯卡上使用PyTorch時(shí)，可以安裝11.6以上版本以及cuDNN 8.1以上版本的計(jì)算包，而在10系列版本的顯卡上，建議優(yōu)先使用2.0版本以前的PyTorch。

下面以CUDA 11.7+cuDNN 8.2.0組合為例，演示完整的PyTorch 2.0 GPU Nvidia運(yùn)行庫(kù)的安裝步驟，其他不同版本CUDA+cuDNN組合的安裝過(guò)程基本一致。