Hi tyhawa, i ran into similar problem like you. Could you please let me know the full configuration of your env. I will try to get my env. config exactly same as you. Thank you so much @tyhawa .
you can grab your env configuration by using this command: python3 -m torch.utils.collect_env