返回首頁

GitHub Repo

ROCm 維護者確認 PyTorch 套件錯配問題已緩解,穩定通道文件仍留有 alpha 版本

ROCm TheRock 維護者於 10 月 16 日將 PyTorch 發布流程的套件錯配問題標記為已修復或緩解。修補加入版本綁定並調整正式分支的版本號,但官方安裝文件仍列出部分 alpha 套件,部署者須核對實際組合。

Thornbury, Walter, 1828-1876 · No restrictions · Image source
zh-Hant

ROCm TheRock 維護者 ScottTodd 於 10 月 16 日關閉一項涉及 PyTorch 發布流程的問題,表示目前已修復或緩解,同時承認發布各階段仍可增加檢查。這次更新確認的是先前修補的處理狀態;相關變更已於 10 月上旬進入 nightly,不能據此推定所有既有穩定套件都已重新發布。問題紀錄

問題出在套件版本與安裝器選擇規則的交互作用。原始回報指出,ROCm 10.1.0 穩定索引放入了帶有 a0 標記的 torchvision 0.29。未指定版本的一般安裝卻取得 torch 2.14 與 torchvision 0.28,執行 ComfyUI 等應用時便可能出現 torchvision::nms 算子不存在的錯誤。套件下載成功,並不代表其中的原生擴充彼此相容。重現紀錄

修補分成兩層:首先,讓 torchvision wheel 的相依中繼資料綁定其建置時使用的 torch 版本,避免解析器自由組合不相容套件;其次,從 PyTorch 正式發布分支建置時,移除套件公開版本的 a0 後綴。前者約束二進位配對,後者處理正式分支產物被當成預發布版本的情況。版本綁定修補、版本號修補

不過,截至查核時,AMD 官方安裝頁面的部分 ROCm 10.1.0 指令仍明確指定 torchvision==0.29.0a0+rocm10.1.0。精確指定版本與不設版本限制的安裝,可能走出不同的解析結果,因此文件範例、索引內容與建置流程必須分別核對。AMD 安裝文件

對維護 AMD GPU 推論環境的工程團隊,這項事件凸顯了套件相依宣告對部署可重現性的影響。依據上述故障機制,驗證環境時應記錄索引與完整版本,並實際載入應用依賴的視覺算子。後續值得追蹤的是穩定索引與文件是否同步,以及發布流程能否自動測試未鎖版的安裝路徑;目前公開證據仍以維護者提供的安裝輸出為主。

來源

  1. PyTorch release workflows can publish alpha versions to stable — Issue #8798
  2. Pin torchvision wheels to the torch version they are built against — PR #8778
  3. Drop a0 suffixes for stable torch package versions — PR #8812
  4. Install PyTorch for ROCm