safetensors 跨平臺(tái)張量存儲(chǔ)與部署1 個(gè)文件在 Windows、Linux、macOS 之間通用【免費(fèi)下載鏈接】safetensorsSimple, safe way to store and distribute tensors項(xiàng)目地址: https://gitcode.com/GitHub_Trending/sa/safetensors模型在 Linux GPU 服務(wù)器上訓(xùn)練完發(fā)版前要在 Windows 筆記本上做一輪推理驗(yàn)證。用 pickle 搬運(yùn)權(quán)重得擔(dān)心版本不匹配和任意代碼執(zhí)行風(fēng)險(xiǎn)換成 safetensors——跨平臺(tái)張量存儲(chǔ)格式——同一個(gè)文件在三臺(tái)機(jī)器上直接可用加載時(shí)不復(fù)制數(shù)據(jù)加載代碼也完全一樣。三平臺(tái)安裝速查平臺(tái)最簡(jiǎn)安裝命令最常見的一個(gè)坑Windowspip install safetensors想從源碼編譯卻沒裝 Visual Studio Build ToolsLinuxpip install safetensors缺 build-essential編譯時(shí)報(bào)找不到 gccmacOSpip install safetensors系統(tǒng) Python 無(wú)寫權(quán)限報(bào) Permission denied注pip 默認(rèn)下載的是預(yù)編譯的 wheel已編譯好的二進(jìn)制包上表命令在三臺(tái)機(jī)器上通用。只有從源碼編譯時(shí)才需要先裝 Rust 工具鏈并把 cargo 加入 PATH?？缙脚_(tái)加載模型的最小代碼import torch from safetensors.torch import save_file # 保存入口 from safetensors import safe_open # 加載入口按上下文打開文件 tensors {w: torch.zeros(1024, 1024)} # 示例張量 save_file(tensors, model.safetensors) # 寫盤頭部 數(shù)據(jù)緩沖區(qū) with safe_open(model.safetensors, frameworkpt, devicecpu) as f: for key in f.keys(): # 遍歷頭部里的全部張量名 print(key, f.get_tensor(key)) # 讀取時(shí)不復(fù)制數(shù)據(jù)這段代碼在三平臺(tái)上原樣可用。文件格式統(tǒng)一為 JSON 頭部 二進(jìn)制數(shù)據(jù)緩沖區(qū)字節(jié)序統(tǒng)一為小端序——即低位字節(jié)排在內(nèi)存前面的存儲(chǔ)方式——任何一臺(tái)機(jī)器存出的文件在另一臺(tái)機(jī)器上按同一套規(guī)則解析即可。踩坑實(shí)錄Windows 上看到 Microsoft Visual C 14.0 or greater is required原因是你在從源碼編譯而機(jī)器上沒有 C 構(gòu)建工具。解法安裝 Visual Studio Build Tools 并勾選桌面 C 開發(fā)組件不想編譯的話跑pip install safetensors直接用預(yù)編譯包即可繞開。Linux 上看到 No such file or directory: gcc 或 command cc failed原因是缺少基礎(chǔ) C/C 構(gòu)建環(huán)境。解法Ubuntu/Debian 跑sudo apt install build-essentialCentOS/RHEL 跑sudo yum groupinstall Development Tools。macOS 上看到 Permission denied原因是系統(tǒng) Python 目錄沒有寫權(quán)限。解法跑pip install --user safetensors或者先用 venv/conda 建隔離環(huán)境再裝后者更干凈。同一文件在生成端能讀、在目標(biāo)端報(bào)錯(cuò)原因是兩端 safetensors 版本不一致。解法各節(jié)點(diǎn)跑pip show safetensors核對(duì)版本號(hào)對(duì)齊后再部署。部署前檢查清單每個(gè)節(jié)點(diǎn)跑pip show safetensors確認(rèn)版本號(hào)一致文件搬運(yùn)后核對(duì)大小與源端一致防止傳輸截?cái)嘞仍谕姹经h(huán)境用 CPU 試加載再進(jìn) GPU 推理鏈路用 venv/conda 隔離環(huán)境不混用系統(tǒng) Python加載后抽查一個(gè)張量的 shape 與數(shù)值別只看沒報(bào)錯(cuò)【免費(fèi)下載鏈接】safetensorsSimple, safe way to store and distribute tensors項(xiàng)目地址: https://gitcode.com/GitHub_Trending/sa/safetensors創(chuàng)作聲明：本文部分內(nèi)容由AI輔助生成（AIGC），僅供參考