簡體   English   中英

Google Vertex AI 預測:為什么 TorchServe 顯示 0 個 GPU?

[英]Google Vertex AI Prediction: Why is TorchServe showing 0 GPUs?

我已將訓練有素的 PyTorch model 部署到 Google Vertex AI 預測端點。 端點運行良好,為我提供了預測,但是當我在日志資源管理器中檢查它的日志時,我看到:

信息 2023-01-11T10:34:53.270885171Z GPU 數量:0

信息 2023-01-11T10:34:53.270888834Z CPU 數量:4

盡管事實上我將端點設置為使用NVIDIA_TESLA_T4作為加速器類型:

顯示選擇 NVIDIA_TESTLA_T4 作為加速器類型的 Google Cloud Console UI 屏幕截圖

為什么日志顯示 0 個 GPU,這是否意味着 TorchServe 沒有利用加速器 GPU?

這是 PyTorch 和 CUDA 的常見 問題。GPU 支持僅在安裝了正確版本的 PyTorch 時啟用,即為 CUDA 編譯的版本。因此建議您使用具有 PyTorch 86831465 功能的圖像

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM