RuntimeError: CUDA error: out of memory

hellloxiaotian / LESRCNN

Lightweight Image Super-Resolution with Enhanced CNN (Knowledge-Based Systems,2020)

217 stars 32 forks source link

Closed buaa1989 closed 3 years ago

buaa1989 commented 3 years ago

我单个GPU（GTX 1660 Ti显卡）训练 x2 model，batchsize改为1，仍然显示 RuntimeError: CUDA error: out of memory。这个正常吗谢谢。

buaa1989 commented 3 years ago

我训练x3 model，batchsize改为1 显示 RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR。谢谢

buaa1989 commented 3 years ago

已解决

aatiqa-ghazali commented 2 years ago

@buaa1989 How did you solve the error?And how much time it took for training

aatiqa-ghazali commented 2 years ago

@buaa1989 你是如何解决错误的？训练花了多少时间