Linux版PyTorch的安全性問題主要包括以下幾個方面:
-
供應鏈安全風險:
- 第三方依賴:PyTorch作為開源軟件,其安全性受第三方依賴影響。2023年,PyTorch官方曾發出警告,指出在PyPI代碼庫上存在與PyTorch的“torchtriton”庫同名的惡意依賴項。這種供應鏈攻擊可能導致敏感數據泄露。
-
遠程命令執行(RCE)漏洞:
- 漏洞CVE-2025-32434:這是一個嚴重的RCE漏洞,影響PyTorch ≤ 2.5.1版本,位于
torch.load()
函數中。當使用weights_only=True
參數時,攻擊者可通過構造惡意模型文件實現遠程代碼執行。該漏洞由研究員Ji’an Zhou發現,并已在PyTorch 2.6.0版本中修復。
-
Linux系統本身的安全性:
- 密碼安全:Linux系統在密碼安全方面存在一些挑戰,但通過實施最小權限原則和定期更新系統,可以顯著提高安全性。
- 更新機制:Linux系統提供了多種更新機制,如
yum update
和pip install --upgrade
,這些機制可以幫助用戶及時更新系統和軟件包,修復已知的安全漏洞。
- 用戶操作建議:使用虛擬環境隔離不同項目的Python環境,避免版本沖突和惡意軟件的傳播。定期檢查和更新PyTorch和其依賴項的版本,確保使用的是最新且安全的版本。
-
權限與認證:
- 禁用root賬戶遠程登錄,使用普通用戶登錄后再切換到root賬戶。
- 修改SSH默認端口以避免暴露在常見攻擊下。
- 啟用公鑰認證,禁用密碼登錄。
-
防火墻配置:
- 使用
firewalld
工具設置合適的入站和出站規則,僅允許必需的網絡流量通過。
-
審計和日志:
- 開啟審計守護進程,配置日志記錄策略,定期檢查和分析日志文件。
-
備份策略:
- 定期備份重要數據和系統配置,以便在數據丟失或系統損壞時恢復。
用戶和管理員應保持警惕,采取適當的安全措施,如定期更新PyTorch版本、使用虛擬環境、配置防火墻等,以提高PyTorch在Linux系統上的安全性。