可以自己編譯,也可以省事用已有的。注意版本對應。
下載地址:
cuDNN(進去就知道下載的CUDA版本了):cuDNN Archive | NVIDIA Developer
CUDA:CUDA Toolkit Archive | NVIDIA Developer ,把下載鏈接里的.com改為.cn,速度可能更快。如果還是慢,登錄賬號后再試試。
onnxruntime(也有現成的Release):microsoft/onnxruntime: ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator (github.com)
【自己編譯】
如果想自己編譯onnxruntime,注意cuDNN與CUDA版本,因為cuDNN更新較慢,所以先看cuDNN支持的CUDA版本。
有關編譯的教程:
onnxruntime (C++/CUDA) 編譯安裝及部署_白色小靴的博客-CSDN博客
windows10 編譯onnxruntime_a little cabbage的博客-CSDN博客_win10 編譯onnxruntime
【用已有的】
onnxruntime——cuda——cudnn:
cuda——顯卡驅動:
CUDA Compatibility :: NVIDIA Data Center GPU Driver Documentation