diff --git a/install.sh b/install.sh new file mode 100644 index 0000000..7417622 --- /dev/null +++ b/install.sh @@ -0,0 +1,4 @@ +#!/bin/bash +conda install gcc gxx 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