Create install.sh

2024-01-17 08:48:27 +08:00
parent 2078ad1177
commit b8c4cbd900
1 changed files with 4 additions and 0 deletions
--- a/install.sh
+++ 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