我寫了以下python代碼
import tvm
ctx = tvm.device("cuda", 0)
print("ctx: ", ctx.exist)
輸出是
ctx: False
但是，我安裝了 cuda 和 cuda 驅動程式。該nvidia-smi結果是，
NVIDIA-SMI 495.44 Driver Version: 495.44 CUDA Version: 11.5
我lspci -vnn | grep VGA用來檢查是否有GPU。結果是
03:00.0 VGA compatible controller [0300]: Matrox Electronics Systems Ltd. Integrated Matrox G200eW3 Graphics Controller [102b:0536] (rev 04) (prog-if 00 [VGA controller])
3b:00.0 VGA compatible controller [0300]: NVIDIA Corporation Device [10de:1e04] (rev a1) (prog-if 00 [VGA controller])
af:00.0 VGA compatible controller [0300]: NVIDIA Corporation Device [10de:1e04] (rev a1) (prog-if 00 [VGA controller])
d8:00.0 VGA compatible controller [0300]: NVIDIA Corporation Device [10de:1e04] (rev a1) (prog-if 00 [VGA controller])
我的系統是。Linux PowerEdge-R740 5.4.0-90-generic #101~18.04.1-Ubuntu我的python版本是3.6。
我不知道為什么 tvm 找不到正確的設備。
uj5u.com熱心網友回復：
一種可能的錯誤來源可能是您在安裝 TVM 時沒有啟用 CUDA 后端支持。
編輯 build/config.cmake 以自定義編譯選項。將 set(USE_CUDA OFF) 更改為 set(USE_CUDA ON) 以啟用 CUDA 后端。對要為其構建的其他后端和庫（OpenCL、RCOM、METAL、VULKAN 等）執行相同操作。
請參考：https : //tvm.apache.org/docs/install/from_source.html#install-from-source
（我會評論，但沒有足夠的聲譽）
轉載請註明出處，本文鏈接：https://www.uj5u.com/gongcheng/364528.html
上一篇：一旦嘗試適合預處理，train_test_split訓練資料上的“IndexError：元組索引超出范圍”
下一篇：缺少從Weka依賴項的匯入