GPU跑pytorch代码的坑

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") 

正常运行，哒哒哒哒哒哒哒哒哒！