runtime error

Exit code: 1. Reason: ========== == CUDA == ========== CUDA Version 12.1.1 Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. This container image and its contents are governed by the NVIDIA Deep Learning Container License. By pulling and using the container, you accept the terms and conditions of this license: https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience. ERROR: Could not find a version that satisfies the requirement torchvision==0.25.0 (from versions: 0.1.6, 0.2.0, 0.16.0+cu121, 0.16.1+cu121, 0.16.2+cu121, 0.17.0+cu121, 0.17.1+cu121, 0.17.2+cu121, 0.18.0+cu121, 0.18.1+cu121, 0.19.0+cu121, 0.19.1+cu121, 0.20.0+cu121, 0.20.1+cu121) ERROR: No matching distribution found for torchvision==0.25.0 Applying ultimate dependency hotfix (PyTorch 2.10.0 for CUDA 12.1)... Traceback (most recent call last): File "/app/train.py", line 12, in <module> subprocess.check_call([ File "/opt/conda/lib/python3.10/subprocess.py", line 369, in check_call raise CalledProcessError(retcode, cmd) subprocess.CalledProcessError: Command '['/opt/conda/bin/python', '-m', 'pip', 'install', 'torch==2.10.0', 'torchvision==0.25.0', 'torchaudio==2.10.0', '--index-url', 'https://download.pytorch.org/whl/cu121', '--upgrade', '--quiet']' returned non-zero exit status 1.

Container logs: