cuda runtime error (38) : no CUDA-capable device is detected

在vscode中训练模型时，出现如下错误：
cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:51

原因在于CUDA没选对，需要在先查看自己有几块显卡：
在terminal中输入python；输入以下代码

import torch
print(torch.cuda.device_count()) #可用GPU数量

查看CUDA版本

nvcc --version # 查看自己的CUDA版本

我的中有两块显卡，在０或１中选一个，写在
model = torch.nn.DataParallel(net).cuda()前面

os.environ['CUDA_VISIBLE_DEVICES'] = '0'
model = torch.nn.DataParallel(net).cuda()

问题解决。

标签：38,nn,capable,torch,cuda,device,no,CUDA
来源： https://blog.csdn.net/Heng1511164937/article/details/117444583