小编Y.J*_*ang的帖子

我正在研究具有一维信号的 CNN。它适用于 CPU 设备完全正常。但是，当我在 GPU 中训练模型时，发生了 CUDA 错误。我在调用os.environ['CUDA_LAUNCH_BLOCKING'] = "1"后设置命令。这样做时，发生了错误而不是错误。虽然 nvidia 文档怀疑硬件问题，但我可以用图像训练其他 CNN，没有任何错误。下面是我在训练模型中加载数据和设置数据的代码。RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILEDcublasCreate(handle)cublasSgemmcublasCreate

    idx = np.arange(len(dataset))  # dataset & label shuffle in once
    np.random.shuffle(idx)

    dataset = dataset[idx]
    sdnn = np.array(sdnn)[idx.astype(int)]        

    train_data, val_data = dataset[:int(0.8 * len(dataset))], dataset[int(0.8 * len(dataset)):]
    train_label, val_label = sdnn[:int(0.8 * len(sdnn))], sdnn[int(0.8 * len(sdnn)):]
    train_set = DataLoader(dataset=train_data, batch_size=opt.batch_size, num_workers=opt.workers)

    for i, data in enumerate(train_set, 0):  # data.shape = [batch_size, 3000(len(signal)), 1(channel)] tensor

        x = data.transpose(1, 2)
        label …