解决报错 RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/sr

原创于 2021-01-25 09:43:55 发布 · 1.8k 阅读

CC 4.0 BY-SA版权

文章标签：

176 篇文章

订阅专栏

本文介绍了一种常见的PyTorch运行时错误——CUDA运行错误，并提供了具体的错误信息及可能的原因。当服务器只有一块GPU且环境变量设置不当时会出现此问题。

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

报错如下：

RuntimeError: cuda runtime error (100) : 
no CUDA-capable device is detected at 
/pytorch/aten/src/THC/THCGeneral.cpp:47

对于单GPU的服务器

os.environ["CUDA_VISIBLE_DEVICES"] = '0'

报以上的错误，可能是由于未设置成 0