diff --git a/install.sh b/install.sh index 7417622..8dadb26 100644 --- a/install.sh +++ b/install.sh @@ -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