attempt to get argmax of an empty sequence

[semel1]

Upon running "main_IRT.py", got an error:

Exception has occurred: ValueError
attempt to get argmax of an empty sequence
  File "D:\colorization\deep-video-prior-master-pytorch\main_IRT.py", line 49, in <module>
    for x in subprocess.Popen("nvidia-smi -q -d Memory | grep -A4 GPU | grep Free", shell=True, stdout=subprocess.PIPE).stdout.readlines()]))

Happened in both Windows 10 and WSL2
Nvidia RTX 3090

[yzxing87]

Seems something went wrong with your GPU driver. Is 'nvidia-smi' normal on your machine? Anyway, you can comment out line 49, and replace it with os.environ["CUDA_VISIBLE_DEVICES"]="0", for simplicity.

[semel1]

Here report from nviia-sm on Windows 10 and WSL2i:

Anyway comment out line 49 only and replace it with os.environ["CUDA_VISIBLE_DEVICES"]="0", throws an error:

  File "main_IRT.py", line 50
    os.environ["CUDA_VISIBLE_DEVICES"] = '0'
     ^
SyntaxError: invalid syntax

However comment out both line 48 and 49 and replace them with os.environ["CUDA_VISIBLE_DEVICES"]="0" - did the job.
Thank you for your prompt responce