Thank you for the reply!
Can I run what you suggested using this code? :
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

If so, I did it the first time. However, the A100 speed was slow, so I lowered the pytorch version to 1.8.0 and cuda version to 11.0.
Should I try again?