Assuming a fresh Anaconda distribution, you can install the dependencies with:
pip3 install torch==1.7.0+cu110 torchvision==0.8.1+cu110
pip install Pillow==8.4.0 visdom==0.1.8.9
pip install opencv-python matplotlib scikit-image
pip install timm tqdm einops=0.4.1 IPython
We ran our experiments with PyTorch 1.7.0, CUDA 11.0, Python 3.6 and Ubuntu 18.04.
Thanks the authors for their works: