ubuntu下安装jittor框架后尝试运行,识别不了GPU

如果您使用 Jittor 时遇到了安装问题、运行错误、输出错误等问题,您可以在 “问答” 区提问。
为了更好地定位您的问题,建议包含以下内容:

  1. 错误信息、
    Failed reason:[f 0517 17:47:25.601990 28 cuda_flags.cc:37] No device found
  2. Jittor 运行的完整 log(建议复制粘贴运行输出,如果可能请不要使用截图)、
    RuntimeError Traceback (most recent call last)
    Cell In[3], line 2
    1 import jittor as jt
    ----> 2 jt.flags.use_cuda = 1

RuntimeError: Wrong inputs arguments, Please refer to examples(help(jt.Flags.use_cuda)).

Types of your inputs are:
self = Flags,
arg = int,

The function declarations are:
void _set_use_cuda(int v)
void _set_use_cuda(bool v)

Failed reason:[f 0517 17:46:52.159971 28 cuda_flags.cc:37] No device found

检查下显卡驱动