@chocolocked did u try updating ur conda environment? i guess it could be because of older packages or version mismatch in case u used pip for installing several libraries.try creating a fresh pytorch env i guess this might solve the issue.