CascadePSP is tested on PyTorch 1.0 though higher version would likely work as well. For installing PyTorch, please refer to the Pytorch website.
We recommend using the anaconda distribution which should contain most of the required dependencies. Other dependencies can be installed by:
pip install progressbar2
conda install cv2You would also need tensorboard for logging.