当前位置: 首页 > 工具软件 > Titan > 使用案例 >

TITAN RTX + cuda9.0 + pytorch报错

秦育
2023-12-01

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

 类似资料: