The version of Pytorch - Githubissues

szymanowiczs / splatter-image

Official implementation of `Splatter Image: Ultra-Fast Single-View 3D Reconstruction' CVPR 2024

https://szymanowiczs.github.io/splatter-image

BSD 3-Clause "New" or "Revised" License

795 stars 54 forks source link

The version of Pytorch #8

Closed sonnefred closed 7 months ago

sonnefred commented 7 months ago

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.

13253318778 commented 7 months ago

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!

szymanowiczs commented 7 months ago

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.

sonnefred commented 7 months ago

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!