This repository has been archived by the owner on Nov 11, 2023. It is now read-only.
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
{{ message }}
-
电脑跑了一晚上，跑到160多的时候报这个错，并且继续训练也会报错，重启未解决，网上搜了一圈按着方法试了试也都不行，不确定原因
卡是1660S+M40 24G，batch_size为5（后来改4没用），环境变量中PYTORCH_CUDA_ALLOC_CONF值为max_split_size_mb:5120
另外我想只用M40 24G训练而不是用1660S训练应该怎么配置，我只知道推理的时候能选，不确定训练的时候该怎么做
Beta Was this translation helpful? Give feedback.
All reactions