Open yxy6217 opened 2 years ago
运行时报错如下 RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/THCBLas.cu:259
Hi @yxy6217, 测试设备的CUDA版本为10.1,CUDNN版本为7.4.2。看报错信息像是torch版本与CUDA版本不匹配,请再检查确认下。
运行时报错如下 RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/THCBLas.cu:259