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
git上的大佬给出的解决方案是cuda版本用cuda10,给出的原因是cuda9不支持最新的显卡架构。
解决方法:
1. 改回NVIDIA 1080Ti
2. 升级cuda版本以及对应的pytorch