Thank you for your help, I finally got it working. I think my anaconda install was corrupted or something. After wasting a lot of time trying to make it work I just uninstalled anaconda, reinstalled it, reinstalled all necessary packages in anaconda and now it works.
I think image transformation is now my bottleneck. Do you have any tricks on windows to make this part faster?
I can see the following trick from the fastai github page… Not sure we have something similar on the windows side:
Note that JPEG decoding can be a bottleneck, particularly if you have a fast GPU. You can optionally install an optimized JPEG decoder as follows (Linux):
conda uninstall --force jpeg libtiff -y
conda install -c conda-forge libjpeg-turbo
CC=“cc -mavx2” pip install --no-cache-dir -U --force-reinstall --no-binary :all: --compile pillow-simd