I’m not sure I understand how to install the matching PyTorch / Cuda 10.1 binary.
Does this look right?:
!pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

The notebook where I’m attempting this is here 3. The test script fails, as do attempts to run the MUNIT training.