Open semel1 opened 2 years ago
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.
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
Upon running "main_IRT.py", got an error:
Happened in both Windows 10 and WSL2 Nvidia RTX 3090