diff --git a/sepal/setup.sh b/sepal/setup.sh new file mode 100644 index 0000000..eb484a9 --- /dev/null +++ b/sepal/setup.sh @@ -0,0 +1,9 @@ +# install mambda +curl -L -O "https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-$(uname)-$(uname -m).sh" +bash yes | Miniforge3-$(uname)-$(uname -m).sh + + +conda create -n torch python=3.11 -y +conda activate torch + +mamba install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia \ No newline at end of file