这个问题卡了一上午,最后发现是版本问题,我的用的笔记本电脑3050,数据预处理的时候不报错,训练的时候报错。在这个帖子RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb-CSDN博客找到解决方法,就是保证30系显卡保证cuda版本>=11.0,
我的python是3.6,所以选择的torch1.7.0+cu110那个就可以运行了。
可能版本太古老了,用官方给的命令会报找不到对应版本的错误。
可以用如下网址离线安装
download.pytorch.org/whl/torchaudio/
说句题外话,就是可能本人有点呆,复现论文的时候要求的是torch1.0,所以就往死里纠结就盯着torch1.0其实尝试不同的版本也可以运行,希望大家不要像我一样呆。
希望能帮到大家