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

您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的 #49

Closed
fatakWang opened this issue Dec 25, 2022 · 2 comments
Closed

您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的 #49

fatakWang opened this issue Dec 25, 2022 · 2 comments

Comments

@fatakWang
Copy link 

    您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的

Originally posted by @fatakWang in #16 (comment)
@fatakWang
Copy link
Author

我已经解决了这个问题，问题出在我的cuda版本安装的是11.6，然而我却按照readme的指示安装了pytorch1.5的版本，然而pytorch1.5只支持cuda10.2的，所以在model.to(cuda)时卡住了，在我使用与cuda11.6对应的pytorch后，并且在目录下修建一个logs的文件夹之后，就可以跑了

@qiqihuang
Copy link

楼上 正解

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
@qiqihuang @fatakWang