Versions of python and pytorch

quancore / social-lstm

Social LSTM implementation in PyTorch

410 stars 248 forks source link

Versions of python and pytorch #9

Closed joanacfcampos closed 5 years ago

joanacfcampos commented 5 years ago

Hi, Would be possible to share the versions of Python, pytorch, numpy, cuda and other relevant packages used in this implementation? I am having getting errors when training a model, so I am wondering if this is related to a mismatch of package versions.

quancore commented 5 years ago

Because I do not have the setup anymore, I cannot say any exact thing. As far as I remember. Pytorch was PyTorch 0.4.0 and python 3.6,

isht7 commented 5 years ago

Please mention in readme that this works for python 3 only