两个错误
1.
Cannot create Cublas handle. Cublas won't be available.
Setting up scale
2.ERROR: Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR
网上查找原因是显卡驱动过高导致。
nvidia-smi查当前驱动
根据经验修改batch_size无果后。然后上网查询解决办法,大部分说是显卡驱动不支持,然后一顿操作卸载重装低版本显卡,无奈我的1050ti只支持390的显卡(ubuntu18.04),最后发现是digits要开启超级权限才可以
然后又报错如上
结绝方法如下
用mnist测试一下效果