如果 MinerU 環境裡檢查 PyTorch 後看到下面這種結果:
|
|
基本可以確定:目前環境安裝的是 CPU 版 PyTorch,所以 MinerU 不會使用 NVIDIA GPU。即使電腦裡有 RTX 4060,只要 Python 環境裡的 torch 是 CPU 版,MinerU 也只能走 CPU。
解決思路很直接:在同一個 .venv 裡,把 torch 和 torchvision 換成 CUDA 版。對於 PyTorch 2.8.0,官方提供 CUDA 12.8 對應版本,可以直接安裝 cu128 輪子。
先確認 NVIDIA 驅動正常
在 PowerShell 執行:
|
|
正常情況下應該能看到類似資訊:
|
|
這裡顯示的 CUDA Version 是目前顯卡驅動最高支援的 CUDA 版本。只要 nvidia-smi 能正常識別顯卡,通常不需要另外安裝完整的 CUDA Toolkit。
如果連 nvidia-smi 都無法識別顯卡,先不要折騰 PyTorch,應該先更新或重新安裝 NVIDIA 驅動。
進入 MinerU 使用的虛擬環境
先進入 MinerU 所在目錄。這裡以 C:\Work\test 為例:
|
|
如果專案使用 .venv,啟用它:
|
|
確認目前 Python 路徑:
|
|
結果應當指向類似路徑:
|
|
這一步很重要。你要替換的是 MinerU 實際使用的那個 Python 環境裡的 PyTorch,而不是系統 Python、Conda 裡的另一個環境,或者 VS Code 隨便選中的解譯器。
卸載 CPU 版 PyTorch
確認 .venv 已啟用後,卸載目前 CPU 版:
|
|
如果你不是用 uv 管理環境,也可以用普通 pip,但同一個環境裡最好不要混亂切換工具。這裡沿用 uv pip。
安裝 CUDA 12.8 版 PyTorch
安裝 PyTorch 2.8.0 對應的 CUDA 12.8 版本:
|
|
PyTorch 官方給出的這組版本對應關係是:
|
|
CUDA 版 PyTorch 下載量可能比較大,通常會達到數 GB。網路慢時需要耐心等一下。
如果 uv 提示已經安裝,但你檢查後仍然是 CPU 版,可以強制重裝:
|
|
安裝後驗證 CUDA 是否可用
執行:
|
|
正確結果應類似:
|
|
最關鍵的是這一行:
|
|
如果這裡仍然是 False,說明 PyTorch 仍然沒有用上 CUDA。常見原因包括:裝錯環境、仍然是 CPU 版 torch、NVIDIA 驅動異常,或者 Python 解譯器不是 MinerU 那個 .venv。
再做一次真實 GPU 運算
只看到 CUDA available: True 還不夠穩妥,可以再跑一個真實 CUDA 張量計算:
|
|
正常結果類似:
|
|
這說明 PyTorch 不只是識別到了顯卡,而且確實能在 RTX 4060 上執行 CUDA 運算。
執行 MinerU 時觀察顯卡
打開第二個 PowerShell 視窗,持續觀察顯卡:
|
|
然後在第一個視窗執行 MinerU:
|
|
如果沒有啟用虛擬環境,也可以直接呼叫 .venv 裡的可執行檔:
|
|
觀察 nvidia-smi,如果滿足下面幾個現象,基本就能確認 MinerU 正在使用 RTX 4060:
- 出現
python.exe。 - 顯存占用增加。
- 推理階段
GPU-Util升高。 - MinerU 結束後顯存釋放。
RTX 4060 和 MinerU 模式怎麼理解
RTX 4060 屬於 NVIDIA 40 系顯卡,也就是 Ada Lovelace 架構,屬於 MinerU 官方支援 GPU 加速的硬體範圍。安裝正確的 CUDA 版 PyTorch 後,普通 mineru 命令就可以使用 GPU。
如果顯式指定:
|
|
它更偏穩定和相容,也適合 CPU 回退或低顯存場景。想使用 hybrid-engine、vlm-engine 這類更依賴 VLM 的模式,就更需要確認 CUDA 版 PyTorch 已經安裝成功。
常見坑
第一,裝錯環境。最常見的問題是你在一個 Python 環境裡安裝了 CUDA 版 PyTorch,但 MinerU 實際執行的是另一個 .venv。所以一定要先看:
|
|
第二,只看驅動,不看 PyTorch。nvidia-smi 正常只能說明驅動識別顯卡,不代表 Python 裡的 PyTorch 一定支援 CUDA。最終還是要看:
|
|
第三,把 CUDA Toolkit 和 PyTorch CUDA 混為一談。多數普通使用場景不需要單獨安裝完整 CUDA Toolkit。只要 NVIDIA 驅動正常,安裝 PyTorch 官方 CUDA 輪子即可。
第四,顯存被別的程式占滿。RTX 4060 8GB 屬於能跑但不寬裕的配置。執行 MinerU 前,最好關掉遊戲、瀏覽器硬體加速、其他 AI 推理程式和占顯存的軟體。
一句話總結
看到:
|
|
就說明 MinerU 目前用不了 RTX 4060。正確做法是在同一個 .venv 裡卸載 CPU 版 torch / torchvision,再安裝:
|
|
驗證結果變成:
|
|
再執行 MinerU,並用 nvidia-smi -l 1 觀察 python.exe、顯存和 GPU 利用率,就能確認 GPU 加速是否真正生效。