From 7e533c6995523748026db98584cdf9855c636368 Mon Sep 17 00:00:00 2001 From: Alex Z Date: Wed, 17 Jan 2024 08:58:19 +0800 Subject: [PATCH] Update install.sh --- install.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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