Devakumar868 commited on
Commit
fc9eb64
·
verified ·
1 Parent(s): 7a94b60

Update setup.sh

Browse files
Files changed (1) hide show
  1. setup.sh +5 -2
setup.sh CHANGED
@@ -1,9 +1,12 @@
1
  #!/usr/bin/env bash
2
 
3
- # 1. Install CUDA-enabled PyTorch
 
 
 
4
  pip install --no-cache-dir \
5
  torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 \
6
  --index-url https://download.pytorch.org/whl/cu118
7
 
8
- # 2. Install all other dependencies
9
  pip install --no-cache-dir -r requirements.txt
 
1
  #!/usr/bin/env bash
2
 
3
+ # Install specific numpy version first to avoid conflicts
4
+ pip install --no-cache-dir numpy==1.23.5
5
+
6
+ # Install CUDA-enabled PyTorch
7
  pip install --no-cache-dir \
8
  torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 \
9
  --index-url https://download.pytorch.org/whl/cu118
10
 
11
+ # Install remaining dependencies
12
  pip install --no-cache-dir -r requirements.txt