PyTorch在Linux系統上確實存在一些安全問題,這些安全問題可能會影響到使用PyTorch的開發者和研究人員。以下是一些關于PyTorch在Linux環境下安全性的關鍵點:
供應鏈安全風險
PyTorch作為開源軟件,其安全性受第三方依賴影響。2023年,PyTorch官方曾發出警告,指出在PyPI代碼庫上存在與PyTorch的“torchtriton”庫同名的惡意依賴項。這種供應鏈攻擊可能導致敏感數據泄露。
遠程命令執行(RCE)漏洞
2025年4月,PyTorch中存在一個嚴重的遠程命令執行(RCE)漏洞CVE-2025-32434,該漏洞位于torch.load()函數中,當使用weights_only=True參數時可能被觸發。這個漏洞可能導致目標系統完全淪陷、敏感數據泄露,并可能在云計算環境中形成橫向滲透。
安全更新和補丁
- 升級PyTorch版本:用戶應立即將PyTorch版本升級至2.6.0或更高版本以修復CVE-2025-32434漏洞。
- 使用pip或conda更新:用戶可以通過pip或conda包管理器來更新PyTorch及其相關庫。
安全配置建議
- 使用虛擬環境:通過創建虛擬環境,用戶可以隔離不同項目的Python環境,避免版本沖突和惡意軟件的傳播。
- 最小權限原則:為運行PyTorch的用戶分配最小必要的權限,避免使用root賬戶運行應用程序。
- 定期檢查和更新:用戶應定期檢查PyTorch和其依賴項的版本,確保使用的是最新且安全的版本。
- 監控和日志:啟用系統日志記錄,并定期檢查日志文件以發現異常行為。
其他安全建議
- 避免使用不知名的AI工具:盡量優先使用網頁的方式去與AI聊,遇到不會的編程點,或者代碼需要改進的,就在網頁上去與AI交互。
- 安全審計:定期進行安全審計,檢查系統配置和代碼中可能存在的安全問題。
總之,雖然PyTorch在Linux系統上存在一些安全問題,但通過采取適當的安全措施,如定期更新、使用虛擬環境、遵循最小權限原則等,可以顯著降低這些風險。