运行 python tools/train_net.py --config-file ./configs/Market1501/sbs_R50.yml MODEL.DEVICE "cuda:0"

JDAI-CV / fast-reid

SOTA Re-identification Methods and Toolbox

Apache License 2.0

3.42k stars 837 forks source link

运行 python tools/train_net.py --config-file ./configs/Market1501/sbs_R50.yml MODEL.DEVICE "cuda:0" #624

Closed 12manan12 closed 2 years ago

12manan12 commented 2 years ago

File "D:\Anaconda3\envs\fast38\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "D:\fast-reid-master.\fastreid\layers\non_local.py", line 45, in forward f = torch.matmul(theta_x, phi_x) RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 这种错误什么原因

18735549442 commented 2 years ago

可能是gpu有占用，试试重启一下电脑再训练。

github-actions[bot] commented 2 years ago

This issue is stale because it has been open for 30 days with no activity.

github-actions[bot] commented 2 years ago

This issue was closed because it has been inactive for 14 days since being marked as stale.

18735549442 commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。您好，我已收到您的邮件。我将尽快给您回复。