Spaces:
Running
on
Zero
Running
on
Zero
# Copyright (c) 2024 Amphion. | |
# | |
# This source code is licensed under the MIT license found in the | |
# LICENSE file in the root directory of this source tree. | |
conda install ffmpeg -y | |
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y | |
pip install -r requirements.txt | |
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ | |