About Pytorch Version - Githubissues

yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021

MIT License

188 stars 34 forks source link

Closed FXLYZ closed 2 years ago

FXLYZ commented 2 years ago

Hi My GPU only support cuda>=11, only support pytorch>=1.7. so i cannt use pytorch=1.1.

yrcong commented 2 years ago

Hi, i remember we didn't use the code (like compiled CUDA) limited to pytorch1.1. It might be easy to shift to pytorch1.7.