Geting this with pytorch 1.8. It also says

you may get better performance by installing NVIDIA’s apex

But this confuses me because I thought apex was integrated into pytorch nowadays?
Could I fix this problem just by upgrading pytorch?