Training Error

[gebaltso]

Hello! When I run the CUDA_VISIBLE_DEVICES=0 python generate_codes.py --classname ffhq --samples 200000 I get the following error:

Traceback (most recent call last):
File "/.../DeltaEdit/generate_codes.py", line 118, in
netG = Generator(args.size, 512, 8).to(device)

RuntimeError: The NVIDIA driver on your system is too old (found version 11040). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

What should I do?

Thanks in advance.