@ptrblck Thank you. I just install the below version and problem solved
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

I would like to ask one question more now I tried different batch sizes but CUDA out of memory remain the same.
RuntimeError: CUDA out of memory. Tried to allocate 290.00 MiB (GPU 0; 7.79 GiB total capacity; 5.51 GiB already allocated; 292.81 MiB free; 5.61 GiB reserved in total by PyTorch)
I am using GeForce RTX 2070. Any hint please?