lilinhai
lilinhai
> Sorry we lost track of the exact environment and package versions. Any pytorch version in (1.5, 2.0) should be fine. The CUDA version is dependent on your pytorch installation....