Update install.sh

This commit is contained in:
Alex Z 2024-01-17 08:58:19 +08:00 committed by GitHub
parent d54a55b680
commit 7e533c6995
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,4 +1,6 @@
#!/bin/bash
conda install gcc gxx ffmpeg cmake
conda install -c conda-forge gcc
conda install -c conda-forge gxx
conda install ffmpeg cmake
conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install -r requirements.txt