Spaces:
Running
Running
File size: 1,168 Bytes
bcc039b |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
#!/bin/bash
# Copyright (c) Meta Platforms, Inc. and affiliates.
#SBATCH --job-name=env_creation
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --gres=gpu:8
#SBATCH --exclusive
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=128
#SBATCH --mem=0
#SBATCH --time=01:00:00
# Exit immediately if a command exits with a non-zero status
set -e
# Start timer
start_time=$(date +%s)
# Get the current date
current_date=$(date +%y%m%d)
# Create environment name with the current date
env_prefix=blt_$current_date
# Create the conda environment
source $CONDA_ROOT/etc/profile.d/conda.sh
conda create -n $env_prefix python=3.11 -y -c anaconda
conda activate $env_prefix
echo "Currently in env $(which python)"
# Install packages
pip install torch==2.5.0 xformers --index-url https://download.pytorch.org/whl/cu121
pip install ninja
pip install --requirement requirements.txt
# End timer
end_time=$(date +%s)
# Calculate elapsed time in seconds
elapsed_time=$((end_time - start_time))
# Convert elapsed time to minutes
elapsed_minutes=$((elapsed_time / 60))
echo "Environment $env_prefix created and all packages installed successfully in $elapsed_minutes minutes!"
|