runtime error

Exit code: 1. Reason: /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:1061: UserWarning: Can't initialize NVML raw_cnt = _raw_device_count_nvml() No CUDA-compatible GPU found. An A100, L4 or L40S GPU is required. Traceback (most recent call last): File "/home/user/app/app.py", line 118, in <module> USED_VRAM_PARAMS = GPU_TO_VRAM_PARAMS[gpu_name] NameError: name 'gpu_name' is not defined

Container logs: