| # 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/ | |