（已解决)RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at

最新推荐文章于 2025-05-28 17:39:12 发布

写代码的二次猿

最新推荐文章于 2025-05-28 17:39:12 发布

阅读量5.9k

点赞数 1

CC 4.0 BY-SA版权

文章标签： pytorch

本文链接：https://blog.csdn.net/imqyt/article/details/124021800

在尝试将代码上传到服务器并运行时，遇到了CUDA错误提示，表明没有检测到CUDA兼容的设备。问题源于代码中设置了错误的CUDA设备ID。解决方案是检查并修改CUDA_VISIBLE_DEVICES环境变量，确保其设置为实际存在的GPU编号，例如改为0。通过这样做，可以避免CUDA初始化时的错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们将我们的代码上传到服务器运行是，遇到下面问题：

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=50 error=100 : no CUDA-capable device is detected
Traceback (most recent call last):
  File "HyperAttentionDTI_main.py", line 185, in <module>
    model = AttentionDTI(hp).cuda()
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 304, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 201, in _apply
    module._apply(fn)
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 223, in _apply
    param_applied = fn(param)
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 304, in <lambda>
    return self._apply(lambda t: t.cuda(device))
  File "/usr/local/lib/python3.7/site-packages/torch/cuda/__init__.py", line 197, in _lazy_init
    torch._C._cuda_init()
RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50

是因为我们的显卡设置错误，因为我们有没有显卡，自己知道，不然不会报错显卡问题

解决办法：

我们看看自己执行的程序代码，好好检查cuda部分，我把我的显卡设置成了6，名我们没那么多显卡可以用，我查看了自己的显卡位置，在0号，所以，我们把下面的代码第一行，设置为0，就可以了！

os.environ["CUDA_VISIBLE_DEVICES"] = "6"
if __name__ == "__main__":
    """select seed"""
    SEED = 1234
    random.seed(SEED)
    torch.manual_seed(SEED)
    torch.cuda.manual_seed_all(SEED)
    # torch.backends.cudnn.deterministic = True