AI开发平台MODELARTS-日志提示"No CUDA-capable device is detected":处理方法

时间:2024-04-30 17:35:55

处理方法

  1. 尽量代码里不要去修改CUDA_VISIBLE_DEVICES变量,用系统默认里面自带的。
  2. 如果必须指定卡ID,需要注意一下1/2/4规格下,指定的卡ID与实际分配的卡ID不匹配的情况。
  3. 如果上述方法还出现了错误,可以去notebook里面调试打印CUDA_VISIBLE_DEVICES变量,或者用以下代码测试一下,查看结果是否返回的是True。
    import torch
    torch.cuda.is_available()
support.huaweicloud.com/trouble-modelarts/modelarts_trouble_0032.html