The program runs and appears
RuntimeError: CUDA unknown error-this may be due to an incorrectly set up
Later it was solved in the following way:
apt-get install nvidia-modprobe