1、问题:
AssertionError: CUDA unavailable, invalid device 0 requested
2、解决方案
主要原因是目前服务器GPU不可以
查看情况:
查看服务器使用情况:nvidia-smi
查看是否有GPU:
import torch
torch.cuda.is_available()
结果:若上面显示False,恭喜您,当前GPU不可用
yolov7训练问题:AssertionError: CUDA unavailable, invalid device 0 requested
最新推荐文章于 2024-03-31 18:19:13 发布