installer

src/mlspace/cli/setup.py

@@ -11,6 +11,12 @@ class SetupMLSpaceCommand(BaseCommand):
     @staticmethod
     def register_subcommand(parser: ArgumentParser):
         _parser = parser.add_parser("setup", help="Setup MLSpace and install all dependencies. Run with `sudo`")
+        _parser.add_argument(
+            "--generate_install_script",
+            help="Adding this argument will just generate the install script. Used in development only.",
+            action="store_true",
+            required=False,
+        )
         _parser.set_defaults(func=setup_mlspace_command_factory)
 
     def execute(self):

src/mlspace/mlspace.py

@@ -5,10 +5,7 @@
 from .dockerfiles import TORCH_GPU
 from .logger import logger
 from .requirements import BASE_REQUIREMENTS
-from getpass import getpass
-import tempfile
-
-from .scripts import install_docker, install_nvidia_docker, install_nvidia_drivers, base_script
+from .scripts import base_script, install_docker, install_nvidia_docker, install_nvidia_drivers
 
 
 class MLSpace:

src/mlspace/scripts/__init__.py

@@ -1,4 +1,4 @@
+from .base import script as base_script
 from .install_docker import script as install_docker
 from .install_nvidia_docker import script as install_nvidia_docker
 from .install_nvidia_drivers import script as install_nvidia_drivers
-from .base import script as base_script

src/mlspace/scripts/install_docker.py

@@ -1,25 +1,25 @@
 script = """
-apt-get remove -y docker docker-engine docker.io containerd runc
-apt-get update
-apt-get install -y \
+sudo apt-get remove -y docker docker-engine docker.io containerd runc
+sudo apt-get update
+sudo apt-get install -y \
     ca-certificates \
     curl \
     gnupg \
     lsb-release
 
-rm -rf /usr/share/keyrings/docker-archive-keyring.gpg
-su - "$USER" -c "curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg"
+sudo rm -rf /usr/share/keyrings/docker-archive-keyring.gpg
+curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg
 
 echo \
   "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu \
-  $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null
+  $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
 
-apt-get update
-apt-get install -y docker-ce docker-ce-cli containerd.io
+sudo apt-get update
+sudo apt-get install -y docker-ce docker-ce-cli containerd.io
 
-groupadd -f docker
-usermod -aG docker $USER
+sudo groupadd -f docker
+sudo usermod -aG docker $USER
 
-systemctl enable docker.service
-systemctl enable containerd.service
+sudo systemctl enable docker.service
+sudo systemctl enable containerd.service
 """

src/mlspace/scripts/install_mlspace.py

@@ -0,0 +1,12 @@
+script = """
+set -e
+
+sudo apt-get update
+
+sudo apt-get install -y python3 python3-pip
+sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 1
+sudo update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
+
+pip install -U pip
+pip install -U mlspace
+"""

src/mlspace/scripts/install_nvidia_docker.py

@@ -3,7 +3,7 @@
    && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - \
    && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
 
-apt-get update
-apt-get install -y nvidia-docker2
-systemctl restart docker
+sudo apt-get update
+sudo apt-get install -y nvidia-docker2
+sudo systemctl restart docker
 """

src/mlspace/scripts/install_nvidia_drivers.py

@@ -1,8 +1,4 @@
 script = """
-#!/usr/bin/env bash
-
-set -e
-
 sudo add-apt-repository -y ppa:graphics-drivers/ppa
 sudo apt-get install -y nvidia-driver-{version}
 """