mohit117:

conda install pytorch torchvision torchaudio cudatoolkit=10.0 -c pytorch


I am not sure if pytorch is available with cuda 10.0.
Can you try with
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch as given in https://pytorch.org/ 5?