AI News HubLIVE
站內改寫2 分鐘閱讀

AgentEnv:開源的大規模智慧體環境執行平臺

AgentEnv(AENV)是一個基於Firecracker微虛擬機器和OCI映象的開源平臺,用於大規模執行智慧體環境,為Kimi K3的強化學習訓練提供支援。它具備快照啟動(<50ms)、增量快照、記憶體氣球、高I/O效能等特性,並提供了CLI和E2B相容API。

來源Hacker News AI作者: handfuloflight

AgentEnv(簡稱AENV)是一個開源的智慧體環境執行平臺,專門為大規模執行智慧體環境而設計,為Kimi K3的強化學習訓練提供動力。該平臺基於Firecracker微虛擬機器技術,結合OCI相容的容器映象,透過overlaybd按需載入,實現了跨物理機的大規模環境部署。本地磁碟作為有界快取,保留熱資料並自動淘汰冷資料,使得映象總大小可以超過磁碟容量,同時保持整個叢集的快速啟動,無需預熱每臺主機。

在效能方面,AENV的獨特之處在於其快照機制。基於快照的環境啟動或恢復時間低於50毫秒,暫停時間低於100毫秒。空閒環境可以迅速釋放CPU和記憶體資源,待新任務到來時再快速恢復,從而顯著降低空閒成本。平臺支援對記憶體和檔案系統進行增量快照,即使在高磁碟修改量下,完成時間也低於100毫秒。正在執行的環境可以分叉(fork)成多個獨立的沙箱,用於並行智慧體工作流。快照可持久化到S3相容的物件儲存或共享分散式檔案系統,防止資料丟失。

在I/O效能方面,AENV透過ublk提供高效能I/O,同時共享主機頁面快取,用於儲存和記憶體快照資料。記憶體氣球技術將可回收的客戶機記憶體歸還給主機,隨著環境執行時間增長和差異變大,依然維持高密度和高效能。

平臺對硬體和軟體有特定要求:需要Linux核心6.8及以上版本,並且擁有/dev/kvm訪問許可權以執行Firecracker微VM。快速部署方面,提供了Ubuntu 24.04的一鍵安裝指令碼和Docker兩種方式。安裝指令碼會同時安裝伺服器和aenv CLI,並將伺服器作為systemd服務啟動。Docker方式則透過docker-setup.sh指令碼準備環境,然後拉取並執行伺服器容器。

使用aenv CLI可以進行模板管理、沙箱生命週期管理(啟動、暫停、恢復、刪除)以及超時設定等操作。例如,透過“aenv pull ubuntu:22.04 --name ubuntu”拉取模板,然後用“aenv start ubuntu”啟動一個沙箱並附加互動式shell。CLI支援表格和JSON兩種輸出格式。

AENV還提供了與E2B相容的HTTP API,只需將E2B_API_URL指向AENV伺服器,即可使用標準的E2B Python或TypeScript SDK,無需修改程式碼。在部署方面,支援Docker Compose和Kubernetes叢集部署,以及從原始碼構建。

該專案採用MIT許可證,目前在GitHub上已獲得超過1100顆星和92次分叉,社群活躍。