zju3dv / SMAP

[ECCV 2020] SMAP: Single-Shot Multi-Person Absolute 3D Pose Estimation
Apache License 2.0
241 stars 37 forks source link

Error with tesh.sh (dapalib fails) #12

Closed minhpvo closed 3 years ago

minhpvo commented 3 years ago

Thanks for this great repo. I'm trying to run the code from pretrained SMAP. I followed the instructions same as Run inference on custom images but I'm getting following error (attached error screenshot) in stage3_root2 test.sh

Capture

My environment is torch1.5.0, torchvision0.6, and gcc 7.5c.

I am looking forward to hear from you!

raypine commented 3 years ago

What's your CUDA version ? CUDA 10.1 seems compatible.

minhpvo commented 3 years ago

I have cuda 10.2. It appears to be a confusion between python and python3.