CUDA driver version is insufficient for CUDA runtime version

[tobeyond]

I tried this:

(jukebox) C:\Users\...\jukebox>python jukebox/sample.py --model=1b_lyrics --name=sample_1b --levels=3 --sample_length_in_seconds=20 --total_sample_length_in_seconds=180 --sr=44100 --n_samples=16 --hop_fraction=0.5,0.5,0.125

And I got:

Using cuda False
THCudaCheck FAIL file=..\torch\csrc\cuda\Module.cpp line=33 error=35 : CUDA driver version is insufficient for CUDA runtime version
Traceback (most recent call last):
  File "jukebox/sample.py", line 275, in <module>
    fire.Fire(run)
  File "C:\Users\...\.conda\envs\jukebox\lib\site-packages\fire\core.py", line 127, in Fire
    component_trace = _Fire(component, args, context, name)
  File "C:\Users\...\.conda\envs\jukebox\lib\site-packages\fire\core.py", line 366, in _Fire
    component, remaining_args)
  File "C:\Users\...\.conda\envs\jukebox\lib\site-packages\fire\core.py", line 542, in _CallCallable
    result = fn(*varargs, **kwargs)
  File "jukebox/sample.py", line 267, in run
    rank, local_rank, device = setup_dist_from_mpi(port=port)
  File "c:\users\...\jukebox\jukebox\utils\dist_utils.py", line 55, in setup_dist_from_mpi
    torch.cuda.set_device(local_rank)
  File "C:\Users\...\.conda\envs\jukebox\lib\site-packages\torch\cuda\__init__.py", line 265, in set_device
    torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at ..\torch\csrc\cuda\Module.cpp:33