训练时报错
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
网上都说是cuda版本问题,我用的cuda是9.0,用cuda10.0就没问题
由于实验室的机器是大家一起用的,改变cuda的版本估计不太现实,也没有试过在docker中安装cuda10.0
但是,将batchsize改小之后,问题解决。
训练时报错
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
网上都说是cuda版本问题,我用的cuda是9.0,用cuda10.0就没问题
由于实验室的机器是大家一起用的,改变cuda的版本估计不太现实,也没有试过在docker中安装cuda10.0
但是,将batchsize改小之后,问题解决。