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 (3,844 star) (790 fork)batch import
help wanted

描述

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

貢獻者指南

技術棧
pythonpytorch
領域
machine learningbackend
議題類型
bug
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
4
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
half day
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
mostly clear
前置要求
Familiarity with PyTorch and CUDAUnderstanding of GPU compute capabilities
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
30
研究方向
This issue requires investigating the CUDA compute capability of the A30 GPU (8.0) and ensuring the installed PyTorch version (1.12.0+cu113) includes kernels for that architecture. Check the PyTorch documentation on CUDA compute capabilities and possibly reinstall a compatible PyTorch build. Also verify the torchserve configuration for GPU device selection.