diff --git a/utils.sh b/utils.sh deleted file mode 100644 index a250acf..0000000 --- a/utils.sh +++ /dev/null @@ -1,84 +0,0 @@ -#!/bin/bash - -# Function to check if PyTorch is installed -check_pytorch_installed() { - python -c "import torch" >/dev/null 2>&1 - return $? -} - -# Function to create virtual environment -create_venv() { - if [ -d "venv" ]; then - echo "Virtual environment already exists. Skipping creation." - source venv/bin/activate - else - echo "Creating virtual environment..." - python3.10 -m venv --system-site-packages venv - source venv/bin/activate - python3.10 -m pip install --upgrade pip - fi -} - -use_venv() { - echo "Connecting to virtual environment..." - source venv/bin/activate -} - -# Function to install build-essential or equivalent -install_build_essentials() { - echo "Checking for build essentials..." - - if [ -f /etc/debian_version ]; then - sudo apt-get update - sudo apt-get install -y build-essential python3.10-dev - - elif [ -f /etc/fedora-release ]; then - if rpm -q gcc &>/dev/null && rpm -q python3.10-devel &>/dev/null; then - echo "Development Tools and Python 3.10 development files are already installed." - else - echo "Installing Development Tools and Python 3.10 development files..." - sudo dnf groupinstall -y "Development Tools" - sudo dnf install -y python3.10-devel - fi - - else - echo "Unsupported operating system. Please install build-essential or equivalent manually." - exit 1 - fi -} - -# Function to install PyTorch in the virtual environment -install_pytorch() { - # Check if PyTorch is installed - if check_pytorch_installed; then - echo "PyTorch is already installed." - else - echo "Installing PyTorch..." - python3.10 -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1 - fi -} - -prepare_env(){ - # Create virtual environment - create_venv - - # Install build essentials - install_build_essentials - - # Install PyTorch in the virtual environment - install_pytorch -} - -clean() { - python3.10 -m pip cache purge -} - -yes_or_no() { - while true; do - read -p "$* [y/n]: " yn - case $yn in - [Yy]*) return 0 ;; - [Nn]*) echo "Aborted" ; return 1 ;; - esac - done -} \ No newline at end of file