OR you can follow the steps below: The code is tested with python 3.9, cuda == 11.3, pytorch == 1.10.1. Additionally dependencies include: h5py kornia torch torchvision omegaconf torchmetrics==0.10.3 ...
The case study relies on a number of external packages. It's often best to start with a tool like conda to build virtual environments and download packages. This can also be done with other virtual ...