File size: 605 Bytes
bab2810
 
02bc3cb
bab2810
02bc3cb
 
bab2810
 
6b024db
b1229b3
bab2810
02bc3cb
b1229b3
 
02bc3cb
bab2810
02bc3cb
bab2810
 
 
 
02bc3cb
bab2810
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#!/bin/bash

echo "πŸš€ Installing Python packages..."

# Install core dependencies from pip
pip install --upgrade pip
pip install --no-cache-dir -r /requirements.txt

# Transformer Engine
pip install transformer-engine[pytorch]==1.12.0

# Apex
git clone https://github.com/NVIDIA/apex
cd apex
CUDA_HOME=/usr/local/cuda pip install -v --no-cache-dir --disable-pip-version-check --no-build-isolation \
  --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" .
cd ..

# MoGe
pip install git+https://github.com/microsoft/MoGe.git

# Launch your app
cd /app
python app.py