search

InhwanBae / GPGraph

Official Code for "Learning Pedestrian Group Representations for Multi-modal Trajectory Prediction (ECCV 2022)"
https://ihbae.com/publication/gpgraph/
MIT License
59 stars 7 forks source link

version problem #1

Closed 12num closed 1 year ago

12num commented 1 year ago

What are the versions of cuda and pytorch you are using, and I did not find your requirements.tx. Thank you very much if you can share it！

InhwanBae commented 1 year ago

Hi @12num!

Thanks for your interest in my work! I did all experiments on the RTX 3090 system(Ubuntu 20.04, PyTorch 1.9.0, CUDA 11.1). However, I checked that this code works well in various versions of PyTorch(1.7, 1.8, 1.9), CUDA(10, 11) and GPU(2080ti, 3090).

12num commented 1 year ago

Thank you very much for your reply, very useful. Thanks again for sharing your work publicly.