install pytorch with cuda Go this : https://pytorch.org/ install cuda Go this : https://developer.nvidia.com/cuda-11-6-2-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local $ wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run $ sudo sh cuda_11.6.2_510.47.03_linux.run CU..
