The version of Pytorch

Question

The version of Pytorch

sonnefred opened this issue 8 months ago · 3 comments

Hi, could you tell me which versions are supported by this project? I used torch 1.8.1, but got this error. I think it's related to the version of Pytorch. Could you give me any hint about this? Thanks.

Answer 1 · 2024-01-05T08:32:35.000Z

Maybe pytorch1.12 can solve your problem，I don't have your problem with 1.12, but I do have a problem with dataset preprocessing!

Answer 2 · 2024-01-13T17:52:40.000Z

Hi @sonnefred - I verified that the code works with:

Python 3.7, Pytorch 1.12.1, CUDA 11.6
Python 3.8, Pytorch 2.1.1, CUDA 12.1

Hopefully this helps. I also updated the instructions in Readme.

Answer 3 · 2024-01-15T03:29:07.000Z

Hi @sonnefred - I verified that the code works with:
* Python 3.7, Pytorch 1.12.1, CUDA 11.6

* Python 3.8, Pytorch 2.1.1, CUDA 12.1
Hopefully this helps. I also updated the instructions in Readme.

Thanks for your reply, and I will have a try!