报错信息：

File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 297, in _conv_forward
    return F.conv1d(input, weight, bias, self.stride,
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)

解决方法：
禁用cuDNN，在训练代码的顶部添加以下代码：

import torch
torch.backends.cudnn.enabled = False

其他说明：
torch.backends.cudnn.benchmark = False 是关闭自动调优。
torch.backends.cudnn.enabled = False 是完全禁用cuDNN。
在遇到cuDNN相关的问题时，首先尝试关闭自动调优，如果问题依然存在，再考虑完全禁用cuDNN。