RuntimeError: cublas runtime error : the GPU program failed to execute at at pytorch/work/aten/src/THC/THCBlas.cu:258
git上的大佬给出的解决方案是cuda版本用cuda10,给出的原因是cuda9不支持最新的显卡架构。
解决方法:
1. 改回NVIDIA 1080Ti
2. 升级cuda版本以及对应的pytorch
博客指出出现RuntimeError,提示GPU程序执行失败。git大佬给出解决方案,因cuda9不支持最新显卡架构,建议使用cuda10。解决方法一是改回NVIDIA 1080Ti,二是升级cuda版本及对应的pytorch。
git上的大佬给出的解决方案是cuda版本用cuda10,给出的原因是cuda9不支持最新的显卡架构。
解决方法:
1. 改回NVIDIA 1080Ti
2. 升级cuda版本以及对应的pytorch
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4005
2545

被折叠的 条评论
为什么被折叠?