Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

START FEATURE EXTRACTION出现错误 #10

Closed
panda-tang opened this issue Dec 2, 2021 · 5 comments
Closed

START FEATURE EXTRACTION出现错误 #10

panda-tang opened this issue Dec 2, 2021 · 5 comments

Comments

@panda-tang
Copy link

作者您好，我运行您的代码之后，在START FEATURE EXTRACTION阶段出现RuntimeError: CUDA error: unknown error，其它地方均无错误，请问我该如何解决它？
image
image
@ttengwang
Copy link
Owner

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本

@panda-tang
Copy link
Author

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本
显卡：NVIDIA GeForce GTX 1050
CUDA Version 10.1.243
Python3.7
pytorch1.71
image

@ttengwang
Copy link
Owner

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本
显卡：NVIDIA GeForce GTX 1050
CUDA Version 10.1.243
Python3.7
pytorch1.71
image

@Tonytom66 谢谢，应该是1050显存只有2GB，显存不够。参考eriklindernoren/PyTorch-YOLOv3#59

@ttengwang
Copy link
Owner

@Tonytom66 请尝试在python video_backbone/TSP/extract_features/extract_features.py 指令后面多加一个参数 --batch-size 这里batch size默认为32，减小batch size后应该可以解决此问题。

@panda-tang
Copy link
Author

@ttengwang 这确实解决了我的问题，非常感谢！！！

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@ttengwang @panda-tang