win10搭配AMD ROCm7.1安裝Pytorch

更新 發佈閱讀 2 分鐘

AMD在2025.11月釋出ROCm 7.1, (聽說)用這個可以跑Pytorch

實在是太新了, 連Google Gemini都不知道怎麼安裝, 問了老半天一直鬼打牆,最後還是自己慢慢搜尋。

以下是我的電腦規格

處理器 AMD Ryzen 7 7700 8-Core Processor 3.80 GHz 顯示卡 AMD Radeon RX 9060 XT (16 GB), AMD Radeon(TM) Graphics (486 MB) 系統類型 64 位元作業系統,x64 型處理器

參考這篇文章:

Use ROCm on Radeon and Ryzen

其實也沒有很複雜,一開始我用Python 3.13安裝失敗,想說奇怪Python不是向下相容嗎? 結果安裝時跳這個錯誤:

ERROR: torch-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl is not a supported wheel on this platform.

查一下才知道cp312就只支援python-3.12,傻傻的重新用一個python-3.12的venv就成功了

raw-image



留言
avatar-img
留言分享你的想法!
avatar-img
OuO OmO的沙龍
2會員
2內容數