cuda、cudnn、zlib 深度學習GPU必配三件套(Windows)

無論用tensorrt,還是onnxruntime部署。這三個都得下載配置,推薦都放到相應的cuda路徑裏(含dll的文件夾、含lib的文件夾)。

推薦先下載tensorrt或者onnxruntime,這樣可以知道所需的cuda、cudnn版本。

1、cuda(以11.8爲例)

下載地址:CUDA Toolkit Archive | NVIDIA 開發者

默認安裝在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

2、cudnn

下載地址:cuDNN Archive | NVIDIA 開發者

解壓後,

dll 放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

lib 放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64

3、zlib

Zlib是cuDNN所需的數據壓縮軟件庫。

下載地址:Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

 解壓後,

zlibwapi.dll 放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

zlibwapi.lib 放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章