PyTorch廣泛使用的開源深度學習框架，在分布式訓練場景下確實存在一些安全漏洞和潛在的安全風險。以下是關于PyTorch分布式安全問題的一些詳細信息：
PyTorch分布式安全問題
- RCE漏洞（CVE-2024-5480）：此漏洞允許攻擊者通過構造特定的惡意請求，在目標節點上執行任意代碼，影響整個訓練環境。
- 反序列化漏洞（CVE-2024-48063）：此漏洞存在于PyTorch 2.4.1及以下版本的分布式RPC框架中，攻擊者可以通過客戶端發送包含惡意方法的RemoteModule實例，觸發反序列化執行任意命令。
防御措施
- 更新PyTorch版本：及時更新到最新版本的PyTorch，以修復已知的安全漏洞。
- 限制訪問：通過防火墻限制RPC服務的IP范圍，避免直接暴露在公網上。
- 加強驗證：對RPC請求的輸入參數進行驗證，防止惡意代碼注入。
- 日志監控：開啟節點行為日志，及時檢測異?；顒?，以便快速響應潛在的安全威脅。
通過采取上述措施，可以在很大程度上降低PyTorch分布式訓練過程中的安全風險，保護用戶數據和模型安全。
