报错原因:
配置了多显卡环境,如果使用了Intel MKL作为BLAS,可能是Intel MKL的浮点数计算功能没有设置正确,可以使用安装CUDA时运行./deviceQuery时标注的0号GPU("Device 0")跑跑测试试试看
解决方法:
export CUDA_VISIBLE_DEVICES=0
然后make runtest
原文地址:https://www.cnblogs.com/laowangxieboke/p/10346853.html
时间: 2024-11-29 11:26:59