Devakumar868 commited on
Commit
55c5dda
·
verified ·
1 Parent(s): 322ba51

Update setup.sh

Browse files
Files changed (1) hide show
  1. setup.sh +7 -1
setup.sh CHANGED
@@ -1,12 +1,18 @@
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
 
 
 
1
  #!/usr/bin/env bash
2
 
3
+ echo "🔧 Starting custom setup..."
4
+
5
+ # Install numpy first to avoid conflicts
6
  pip install --no-cache-dir numpy==1.23.5
7
 
8
+ echo "📦 Installing CUDA-enabled PyTorch..."
9
  # Install CUDA-enabled PyTorch
10
  pip install --no-cache-dir \
11
  torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 \
12
  --index-url https://download.pytorch.org/whl/cu118
13
 
14
+ echo "📋 Installing remaining dependencies..."
15
  # Install remaining dependencies
16
  pip install --no-cache-dir -r requirements.txt
17
+
18
+ echo "✅ Setup complete!"