报错信息
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.
解决方法：
原因：选择GPU设备时torch没有识别到。
解决方法：
加一句torch.cuda.device_count()代码
import torch
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
torch.cuda.device_count()
真的是烦死了，一上午时间，中午也没吃饭，一直在改这个破错误！现在终于跑起来了！！！！！！
发布者：admin，转转请注明出处：http://www.yc00.com/web/1738332666a4038274.html
评论列表（0条）