pytorch/serve

RuntimeError: CUDA error: no kernel image is available for execution on the device

Open

#1,754 建立於 2022年7月22日

在 GitHub 查看
 (7 留言) (0 反應) (0 負責人)Java (790 fork)batch import
help wanted

倉庫指標

Star
 (3,844 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

I am deploying a model using torchserve. It was easily deployed on Tesla K80 GPU. But now when i shifted it to newer GPU Nvidia A30. I am getting this error

torch_version: '1.12.0+cu113' cudnn_version: 8302

image

PFA Nvidia-smi Screenshot

image

貢獻者指南

研究方向
驗證Nvidia A30的計算能力（sm 80），並確保安裝的PyTorch版本包含針對該架構的預編譯內核。檢查PyTorch wheel是否使用支持A30的CUDA 11.6+構建。如果沒有，升級PyTorch到支持sm 80的版本或從源碼重新編譯並設置合適的編譯標誌。
技術棧
pythonpytorch
領域
backendinfrastructure
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
半天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonGit
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
55