I don't have enough equipment. so I only run on my CPU
you have to check right version matching with your cuda and OS
Pytorch_install_version
conda create --name deeplearning_torch
I use interpreter belong to Anaconda
blas 1.0 mkl
ca-certificates 2021.10.26 haa95532_2
certifi 2021.10.8 py38haa95532_0
cudatoolkit 10.2.89 h74a9793_1
freetype 2.10.4 hd328e21_0
intel-openmp 2021.4.0 haa95532_3556
jpeg 9d h2bbff1b_0
libpng 1.6.37 h2a8f88b_0
libtiff 4.2.0 hd0e1b90_0
libwebp 1.2.0 h2bbff1b_0
lz4-c 1.9.3 h2bbff1b_1
mkl 2021.4.0 haa95532_640
mkl-service 2.4.0 py38h2bbff1b_0
mkl_fft 1.3.1 py38h277e83a_0
mkl_random 1.2.2 py38hf11a4ad_0
ninja 1.10.2 py38h559b2a2_3
numpy 1.21.2 py38hfca59bb_0
numpy-base 1.21.2 py38h0829f74_0
olefile 0.46 pyhd3eb1b0_0
openssl 1.1.1l h2bbff1b_0
pillow 8.4.0 py38hd45dc43_0
pip 21.2.2 py38haa95532_0
python 3.8.12 h6244533_0
pytorch 1.5.1 py3.8_cuda102_cudnn7_0 pytorch
setuptools 58.0.4 py38haa95532_0
six 1.16.0 pyhd3eb1b0_0
sqlite 3.36.0 h2bbff1b_0
tk 8.6.11 h2bbff1b_0
torchvision 0.6.1 py38_cu102 pytorch
vc 14.2 h21ff451_1
vs2015_runtime 14.27.29016 h5e58377_2
wheel 0.37.0 pyhd3eb1b0_1
wincertstore 0.2 py38haa95532_2
xz 5.2.5 h62dcd97_0
zlib 1.2.11 h62dcd97_4
zstd 1.4.9 h19a0ad4_0