ToTensor is missing from kornia

$USER@koeln:/netscratch/$USER/FR-template-pytorch$ /$USER@koeln:/netscratch/$USER/FR-template-pytorch$ ./run_slurm_imagenet.sh srun: job 102996 queued and waiting for resources srun: job 102996 has been allocated resources pyxis: creating container filesystem ... pyxis: starting container ... Traceback (most recent call last): File "./src/multi-gpu.py", line 19, in from model.resnet50 import PL_ResNet50 File "/netscratch/$USER/FR-template-pytorch/src/model/resnet50.py", line 10, in from kornia.augmentation import RandomResizedCrop, RandomHorizontalFlip, CenterCrop, RandomCrop, Normalize, ToTensor ImportError: cannot import name 'ToTensor' srun: error: geneva: task 0: Exited with exit code 1 srun: Terminating job step 102996.0 $USER@koeln:/netscratch/$USER/FR-template-pytorch$ grep kornia requirements.txt kornia==0.4.0