RuntimeError: scan failed on 2nd step: cudaErrorInvalidValue:invalid argument

zju3dv / SMAP

[ECCV 2020] SMAP: Single-Shot Multi-Person Absolute 3D Pose Estimation

Apache License 2.0

241 stars 37 forks source link

RuntimeError: scan failed on 2nd step: cudaErrorInvalidValue:invalid argument #34

Closed visonpon closed 3 years ago

visonpon commented 3 years ago

Hi, after successfully installed all the dependence, I tried to test on my own images, but got an error like above, and the detail error information like: pred_body2_2d = dapalib.connect(...) RuntimeError: scan failed on 2nd step: cudaErrorInvalidValue:invalid argument

my environment: pytorch1.3 cuda10.1 python3.7