Here is my revision history in Anaconda. Rev 10 is for 1.7, and Rev 11 is for 1.7.1

2020-12-01 15:15:19  (rev 10)
cudatoolkit  {11.0.221 (defaults/win-64) -> 11.0.3 (conda-forge/win-64)}
pytorch  {1.8.0.dev20201021 (pytorch-nightly/win-64) -> 1.7.0 (pytorch/win-64)}
torchvision  {0.8.0.dev20201021 (pytorch-nightly/win-64) -> 0.8.1 (pytorch/win-64)}
+torchaudio-0.7.0 (pytorch/win-64)
2020-12-10 22:57:52  (rev 11)
ca-certificates  {2020.11.8 (conda-forge/win-64) -> 2020.12.5 (conda-forge/win-64)}
certifi  {2020.11.8 (conda-forge/win-64) -> 2020.12.5 (conda-forge/win-64)}
pytorch  {1.7.0 (pytorch/win-64) -> 1.7.1 (pytorch/win-64)}
torchaudio  {0.7.0 (pytorch/win-64) -> 0.7.2 (pytorch/win-64)}
torchvision  {0.8.1 (pytorch/win-64) -> 0.2.2 (pytorch/noarch)}
+six-1.15.0 (conda-forge/noarch)

For whatever reason I see “noarch” in regards to torchvision. I used the command:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
It’s worth mentioning I’ve been modifying the code a little bit in these modules mentioned in the original question, incase that has something to do with it.
As a followup, since I have an RTX 3080 GPU and am using 16 bit precision with torch.cuda.amp.autocast(), is it worth it to upgrade to 1.7.1. I read something about it being faster. Anyway thanks for the help