[英]When using python tvm, No CUDA-capable device is detected
我写了以下 python 代码
import tvm
ctx = tvm.device("cuda", 0)
print("ctx: ", ctx.exist)
output 是
ctx: False
但是,我安装了 cuda 和 cuda 驱动程序。 nvidia-smi
结果是,
NVIDIA-SMI 495.44 Driver Version: 495.44 CUDA Version: 11.5
我使用lspci -vnn | grep VGA
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 找不到正确的设备。
一种可能的错误来源可能是您在安装 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
(我会评论,但没有足够的声誉)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.