眾所周知,嘗試訓練深度神經網路或基本上任何型別的 AI 模型都需要大量的計算能力。在 CPU 上執行此操作可能需要幾天時間,具體取決于專案。
我有一個 AMD GPU,特別是 GPU 的型號名稱是“RX5700”。對于 NVIDIA GPU,有可以輕松用于 GPU 計算的 CUDA,但是,我找不到 AMD 架構的替代方案。我知道有 ROCm,但它只適用于非常特殊的 GPU 系列和 Ubuntu 作業系統。
有沒有可以在 Windows 上使用 RX 系列 GPU 進行 GPU 計算的平臺?
感謝您的回答。
uj5u.com熱心網友回復:
當我擁有 Radeon 6700XT 時,我能夠利用 Microsoft 的 DirectML。它不如 NVIDIA/CUDA/TensorFlow 等。盡管它仍然比在 CPU 上訓練要好得多。盡管它仍然使用舊版本的 TensorFlow,但它的局限性很大。
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/420784.html
標籤:
上一篇:ApacheHttpClient:第一次呼叫非常慢,隨后呼叫非常快
下一篇:Rust和C 的性能差異
