GPU跑代码的坑
RuntimeError: CUDA error: invalid device ordinal
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
很疑惑，我自己明明有一个集显，一个独显，在任务管理器中也显示两个，如下图
然后我使用代码如下
device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
就报错了！很无语，以为电脑 j j 了，问客服，客服说只有在玩游戏的时候才会工作。哦，我买了个啥。
后来我打开 cmd 之后，使用
nvidia-smi 查看下我的GPU,如下
发现我的rtx3060 显卡就是 == cuda:0== ,将代码改回如下
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
正常运行，哒哒哒哒哒哒哒哒哒！