这个问题源于最近在pycharm中跑代码时报错“RuntimeError:cuda runtime error(100) : no CUDA-capable device is detected at…”出现的。看了不少博客说要重装,还好我没有!
加油打工人
1、问题出现:
运行python代码至“torch.cuda.set_device(gpu_ids[0]):”报错如下:
显示我的CUDA有问题,说明我的nvidia驱动没被检测到。
之后我在pycharm里
print(torch.cuda.is_available()) #返回了false!!!!
返回了fa