jetson-examples 0.1.8__py3-none-any.whl → 0.1.9__py3-none-any.whl

reComputer/scripts/comfyui/LICENSE

@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) [2024] [Seeed Studio]
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+MIT License
+
+Copyright (c) [2024] [Seeed Studio]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

reComputer/scripts/comfyui/README.md

@@ -1,127 +1,127 @@
-# Jetson-Example: Run ComfyUI (Stable Diffusion GUI) on NVIDIA Jetson Orin 🚀
-
-## One-Click Quick Deployment of Plug-and-Play Stable Diffusion GUI
-<p align="center">
-  <img src="images/comfyui.png" alt="comfyui">
-</p>
-
-## **Introduction** 📘
-[ComfyUI](https://github.com/comfyanonymous/ComfyUI) will let you design and execute advanced stable diffusion pipelines using a graph/nodes/flowchart based interface. 
-In this project, you can quickly deploy ComfyUI on Nvidia Jetson Orin devices with one click. 
-
-
-
-## **Key Features**:
-- **One-click installation and configuration support for Nvidia Jetson Orin devices.**
-- **GPU acceleration to optimize the performance of stable diffusion pipelines.**
-- Nodes/graph/flowchart interface to experiment and create complex Stable Diffusion workflows without needing to code anything.
-- Fully supports SD1.x, SD2.x, [SDXL](https://comfyanonymous.github.io/ComfyUI_examples/sdxl/), [Stable Video Diffusion](https://comfyanonymous.github.io/ComfyUI_examples/video/), [Stable Cascade](https://comfyanonymous.github.io/ComfyUI_examples/stable_cascade/), [SD3](https://comfyanonymous.github.io/ComfyUI_examples/sd3/) and [Stable Audio](https://comfyanonymous.github.io/ComfyUI_examples/audio/)
-- [Flux](https://comfyanonymous.github.io/ComfyUI_examples/flux/)
-- Asynchronous Queue system
-- Many optimizations: Only re-executes the parts of the workflow that changes between executions.
-- Smart memory management: can automatically run models on GPUs with as low as 1GB vram.
-
-  For other features, please refer to the original project [ComfyUI](https://github.com/comfyanonymous/ComfyUI).
-
-  Workflow examples can be found on the [Examples page](https://comfyanonymous.github.io/ComfyUI_examples/)
-
-
-### Get a Jetson Orin Device 🛒
-| Device Model | Description | Link |
-|--------------|-------------|------|
-| Jetson Orin Nano Dev Kit, Orin Nano 8GB, 40TOPS | Developer kit for NVIDIA Jetson Orin Nano | [Buy Here](https://www.seeedstudio.com/NVIDIAr-Jetson-Orintm-Nano-Developer-Kit-p-5617.html) |
-| reComputer J4012, powered by Orin NX 16GB, 100 TOPS | Embedded computer powered by Orin NX | [Buy Here](https://www.seeedstudio.com/reComputer-J4012-p-5586.html) |
-
-## **Quickstart** ⚡
-
-### Modify Docker Daemon Configuration (Optional)
-To enhance the experience of quickly loading models in Docker, you need to add the following content to the `/etc/docker/daemon.json` file:
-
-```json
-{
-  "default-runtime": "nvidia",
-  "runtimes": {
-    "nvidia": {
-      "path": "nvidia-container-runtime",
-      "runtimeArgs": []
-    }
-  },
-  "storage-driver": "overlay2",
-  "data-root": "/var/lib/docker",
-  "log-driver": "json-file",
-  "log-opts": {
-    "max-size": "100m",
-    "max-file": "3"
-  },
-  "no-new-privileges": true,
-  "experimental": false
-}
-```
-
-After modifying the `daemon.json` file, you need to restart the Docker service to apply the configuration:
-
-```sh
-sudo systemctl restart docker
-```
-
-### **Installation via PyPI (Recommended)** 🐍
-1. Install the package:
-    ```sh
-    pip install jetson-examples
-    ```
-
-2. Restart your reComputer:
-    ```sh
-    sudo reboot
-    ```
-
-3. Run ComfyUI with one command:
-    ```sh
-    reComputer run comfyui
-    ```
-- **Input Dir**: Mount the input directory in Docker to the host directory `~/ComfyUI/input`.
-- **Output Dir**: Mount the output directory in Docker to the host directory `~/ComfyUI/output`.
-
-- **Models Dir**: Mount the models directory in Docker to the host directory `~/ComfyUI/models`.
-
-
-## **For more tutorials** 🔧
-- [ComfyUI Basic Tutorial VN](https://comfyanonymous.github.io/ComfyUI_tutorial_vn/)
-- [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
-- [ComfyUI Examples](https://comfyanonymous.github.io/ComfyUI_examples/)
-- [Comfy Org](https://www.comfy.org/)
-
-
-## **Shortcuts**
-
-| Keybind                            | Explanation                                                                                                        |
-|------------------------------------|--------------------------------------------------------------------------------------------------------------------|
-| Ctrl + Enter                       | Queue up current graph for generation                                                                              |
-| Ctrl + Shift + Enter               | Queue up current graph as first for generation                                                                     |
-| Ctrl + Z/Ctrl + Y                  | Undo/Redo                                                                                                          |
-| Ctrl + S                           | Save workflow                                                                                                      |
-| Ctrl + O                           | Load workflow                                                                                                      |
-| Ctrl + A                           | Select all nodes                                                                                                   |
-| Alt + C                            | Collapse/uncollapse selected nodes                                                                                 |
-| Ctrl + M                           | Mute/unmute selected nodes                                                                                         |
-| Ctrl + B                           | Bypass selected nodes (acts like the node was removed from the graph and the wires reconnected through)            |
-| Delete/Backspace                   | Delete selected nodes                                                                                              |
-| Ctrl + Backspace                   | Delete the current graph                                                                                           |
-| Space                              | Move the canvas around when held and moving the cursor                                                             |
-| Ctrl/Shift + Click                 | Add clicked node to selection                                                                                      |
-| Ctrl + C/Ctrl + V                  | Copy and paste selected nodes (without maintaining connections to outputs of unselected nodes)                     |
-| Ctrl + C/Ctrl + Shift + V          | Copy and paste selected nodes (maintaining connections from outputs of unselected nodes to inputs of pasted nodes) |
-| Shift + Drag                       | Move multiple selected nodes at the same time                                                                      |
-| Ctrl + D                           | Load default graph                                                                                                 |
-| Alt + `+`                          | Canvas Zoom in                                                                                                     |
-| Alt + `-`                          | Canvas Zoom out                                                                                                    |
-| Ctrl + Shift + LMB + Vertical drag | Canvas Zoom in/out                                                                                                 |
-| Q                                  | Toggle visibility of the queue                                                                                     |
-| H                                  | Toggle visibility of history                                                                                       |
-| R                                  | Refresh graph                                                                                                      |
-| Double-Click LMB                   | Open node quick search palette                |
-
-
-## License
-
-This project is licensed under the GNU General Public License v3.0
+# Jetson-Example: Run ComfyUI (Stable Diffusion GUI) on NVIDIA Jetson Orin 🚀
+
+## One-Click Quick Deployment of Plug-and-Play Stable Diffusion GUI
+<p align="center">
+  <img src="images/comfyui.png" alt="comfyui">
+</p>
+
+## **Introduction** 📘
+[ComfyUI](https://github.com/comfyanonymous/ComfyUI) will let you design and execute advanced stable diffusion pipelines using a graph/nodes/flowchart based interface. 
+In this project, you can quickly deploy ComfyUI on Nvidia Jetson Orin devices with one click. 
+
+
+
+## **Key Features**:
+- **One-click installation and configuration support for Nvidia Jetson Orin devices.**
+- **GPU acceleration to optimize the performance of stable diffusion pipelines.**
+- Nodes/graph/flowchart interface to experiment and create complex Stable Diffusion workflows without needing to code anything.
+- Fully supports SD1.x, SD2.x, [SDXL](https://comfyanonymous.github.io/ComfyUI_examples/sdxl/), [Stable Video Diffusion](https://comfyanonymous.github.io/ComfyUI_examples/video/), [Stable Cascade](https://comfyanonymous.github.io/ComfyUI_examples/stable_cascade/), [SD3](https://comfyanonymous.github.io/ComfyUI_examples/sd3/) and [Stable Audio](https://comfyanonymous.github.io/ComfyUI_examples/audio/)
+- [Flux](https://comfyanonymous.github.io/ComfyUI_examples/flux/)
+- Asynchronous Queue system
+- Many optimizations: Only re-executes the parts of the workflow that changes between executions.
+- Smart memory management: can automatically run models on GPUs with as low as 1GB vram.
+
+  For other features, please refer to the original project [ComfyUI](https://github.com/comfyanonymous/ComfyUI).
+
+  Workflow examples can be found on the [Examples page](https://comfyanonymous.github.io/ComfyUI_examples/)
+
+
+### Get a Jetson Orin Device 🛒
+| Device Model | Description | Link |
+|--------------|-------------|------|
+| Jetson Orin Nano Dev Kit, Orin Nano 8GB, 40TOPS | Developer kit for NVIDIA Jetson Orin Nano | [Buy Here](https://www.seeedstudio.com/NVIDIAr-Jetson-Orintm-Nano-Developer-Kit-p-5617.html) |
+| reComputer J4012, powered by Orin NX 16GB, 100 TOPS | Embedded computer powered by Orin NX | [Buy Here](https://www.seeedstudio.com/reComputer-J4012-p-5586.html) |
+
+## **Quickstart** ⚡
+
+### Modify Docker Daemon Configuration (Optional)
+To enhance the experience of quickly loading models in Docker, you need to add the following content to the `/etc/docker/daemon.json` file:
+
+```json
+{
+  "default-runtime": "nvidia",
+  "runtimes": {
+    "nvidia": {
+      "path": "nvidia-container-runtime",
+      "runtimeArgs": []
+    }
+  },
+  "storage-driver": "overlay2",
+  "data-root": "/var/lib/docker",
+  "log-driver": "json-file",
+  "log-opts": {
+    "max-size": "100m",
+    "max-file": "3"
+  },
+  "no-new-privileges": true,
+  "experimental": false
+}
+```
+
+After modifying the `daemon.json` file, you need to restart the Docker service to apply the configuration:
+
+```sh
+sudo systemctl restart docker
+```
+
+### **Installation via PyPI (Recommended)** 🐍
+1. Install the package:
+    ```sh
+    pip install jetson-examples
+    ```
+
+2. Restart your reComputer:
+    ```sh
+    sudo reboot
+    ```
+
+3. Run ComfyUI with one command:
+    ```sh
+    reComputer run comfyui
+    ```
+- **Input Dir**: Mount the input directory in Docker to the host directory `~/ComfyUI/input`.
+- **Output Dir**: Mount the output directory in Docker to the host directory `~/ComfyUI/output`.
+
+- **Models Dir**: Mount the models directory in Docker to the host directory `~/ComfyUI/models`.
+
+
+## **For more tutorials** 🔧
+- [ComfyUI Basic Tutorial VN](https://comfyanonymous.github.io/ComfyUI_tutorial_vn/)
+- [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
+- [ComfyUI Examples](https://comfyanonymous.github.io/ComfyUI_examples/)
+- [Comfy Org](https://www.comfy.org/)
+
+
+## **Shortcuts**
+
+| Keybind                            | Explanation                                                                                                        |
+|------------------------------------|--------------------------------------------------------------------------------------------------------------------|
+| Ctrl + Enter                       | Queue up current graph for generation                                                                              |
+| Ctrl + Shift + Enter               | Queue up current graph as first for generation                                                                     |
+| Ctrl + Z/Ctrl + Y                  | Undo/Redo                                                                                                          |
+| Ctrl + S                           | Save workflow                                                                                                      |
+| Ctrl + O                           | Load workflow                                                                                                      |
+| Ctrl + A                           | Select all nodes                                                                                                   |
+| Alt + C                            | Collapse/uncollapse selected nodes                                                                                 |
+| Ctrl + M                           | Mute/unmute selected nodes                                                                                         |
+| Ctrl + B                           | Bypass selected nodes (acts like the node was removed from the graph and the wires reconnected through)            |
+| Delete/Backspace                   | Delete selected nodes                                                                                              |
+| Ctrl + Backspace                   | Delete the current graph                                                                                           |
+| Space                              | Move the canvas around when held and moving the cursor                                                             |
+| Ctrl/Shift + Click                 | Add clicked node to selection                                                                                      |
+| Ctrl + C/Ctrl + V                  | Copy and paste selected nodes (without maintaining connections to outputs of unselected nodes)                     |
+| Ctrl + C/Ctrl + Shift + V          | Copy and paste selected nodes (maintaining connections from outputs of unselected nodes to inputs of pasted nodes) |
+| Shift + Drag                       | Move multiple selected nodes at the same time                                                                      |
+| Ctrl + D                           | Load default graph                                                                                                 |
+| Alt + `+`                          | Canvas Zoom in                                                                                                     |
+| Alt + `-`                          | Canvas Zoom out                                                                                                    |
+| Ctrl + Shift + LMB + Vertical drag | Canvas Zoom in/out                                                                                                 |
+| Q                                  | Toggle visibility of the queue                                                                                     |
+| H                                  | Toggle visibility of history                                                                                       |
+| R                                  | Refresh graph                                                                                                      |
+| Double-Click LMB                   | Open node quick search palette                |
+
+
+## License
+
+This project is licensed under the GNU General Public License v3.0

reComputer/scripts/comfyui/clean.sh

@@ -1,7 +1,9 @@
-#!/bin/bash
-CONTAINER_NAME="comfyui"
-IMAGE_NAME="yaohui1998/comfyui"
-
-sudo docker stop $CONTAINER_NAME
-sudo docker rm $CONTAINER_NAME
-sudo docker rmi $IMAGE_NAME
+#!/bin/bash
+CONTAINER_NAME="comfyui"
+IMAGE_NAME="yaohui1998/comfyui"
+
+sudo docker stop $CONTAINER_NAME
+sudo docker rm $CONTAINER_NAME
+sudo docker rmi $IMAGE_NAME
+
+sudo rm -r /home/$USER/reComputer/ComfyUI

reComputer/scripts/comfyui/config.yaml

@@ -1,29 +1,30 @@
-allowed_l4t_versions:
-  - 35.3.1 # Add your version here
-required_disk_space: 30  # in GB
-min_mem_gb: 15
-min_swap_gb: 6
-nvidia_jetson_package: "nvidia-jetpack"
-packages:
-  #- "ros-noetic-ros-base"
-  #- "flask"
-docker:
-  desired_daemon_json: |
-    {
-      "default-runtime": "nvidia",
-      "runtimes": {
-        "nvidia": {
-          "path": "nvidia-container-runtime",
-          "runtimeArgs": []
-        }
-      },
-      "storage-driver": "overlay2",
-      "data-root": "/var/lib/docker",
-      "log-driver": "json-file",
-      "log-opts": {
-        "max-size": "100m",
-        "max-file": "3"
-      },
-      "no-new-privileges": true,
-      "experimental": false
-    }
+# The tested JetPack versions.
+ALLOWED_L4T_VERSIONS:
+  - 35.3.1
+  - 35.4.1
+  - 35.5.0
+REQUIRED_DISK_SPACE: 30  # in GB
+REQUIRED_MEM_SPACE: 15
+PACKAGES:
+  - nvidia-jetpack
+DOCKER: 
+  ENABLE: true
+  DAEMON: |
+    {
+      "default-runtime": "nvidia",
+      "runtimes": {
+        "nvidia": {
+          "path": "nvidia-container-runtime",
+          "runtimeArgs": []
+        }
+      },
+      "storage-driver": "overlay2",
+      "data-root": "/var/lib/docker",
+      "log-driver": "json-file",
+      "log-opts": {
+        "max-size": "100m",
+        "max-file": "3"
+      },
+      "no-new-privileges": true,
+      "experimental": false
+    }

reComputer/scripts/comfyui/init.sh

@@ -1,163 +1,9 @@
-#!/bin/bash
-#set color value
-RED=$(tput setaf 1)
-GREEN=$(tput setaf 2)
-YELLOW=$(tput setaf 3)
-BLUE=$(tput setaf 4)
-MAGENTA=$(tput setaf 5)
-CYAN=$(tput setaf 6)
-RESET=$(tput sgr0)
-
-echo "${CYAN}This script will install the necessary packages and configurations for running comfyui on a Jetson.${RESET}"
-
-# Install yq for parsing YAML files
-sudo apt-get update
-sudo apt-get install -y jq
-# Read configuration
-CURRENT_DIR="comfyui"
-CONFIG_FILE="./jetson-examples/reComputer/scripts/${CURRENT_DIR}/config.yaml"
-ALLOWED_L4T_VERSIONS=$(yq -r '.allowed_l4t_versions[]' $CONFIG_FILE)
-ALLOWED_L4T_VERSIONS_ARRAY=($ALLOWED_L4T_VERSIONS)
-REQUIRED_DISK_SPACE=$(yq -r '.required_disk_space' $CONFIG_FILE)
-MIN_MEM_GB=$(yq -r '.min_mem_gb' $CONFIG_FILE)
-MIN_SWAP_GB=$(yq -r '.min_swap_gb' $CONFIG_FILE)
-NVIDIA_JETSON_PACKAGE=$(yq -r '.nvidia_jetson_package' $CONFIG_FILE)
-PACKAGES=$(yq -r '.packages[]' $CONFIG_FILE)
-DESIRED_DAEMON_JSON=$(yq -r '.docker.desired_daemon_json' $CONFIG_FILE)
-CURRENT_DISK_SPACE=$(df -BG --output=avail / | tail -1 | sed 's/[^0-9]*//g')
-MEM_GB=$(free -g | awk '/^Mem:/{print $2}')
-SWAP_GB=$(free -g | awk '/^Swap:/{print $2}')
-
-echo "${MAGENTA}Allowed L4T versions:${RESET} ${GREEN}$ALLOWED_L4T_VERSIONS ${RESET}"
-echo "${MAGENTA}Required disk space:  ${GREEN}${REQUIRED_DISK_SPACE}G ${RESET}"
-echo "${MAGENTA}Minimum memory:  ${GREEN}${MIN_MEM_GB}G ${RESET}"
-echo "${MAGENTA}Minimum swap:  ${GREEN}${MIN_SWAP_GB}G ${RESET}" 
-echo "${MAGENTA}NVIDIA Jetson package:${RESET}  ${GREEN}$NVIDIA_JETSON_PACKAGE ${RESET}"
-echo "${MAGENTA}Additional packages: ${RESET} ${GREEN}$PACKAGES ${RESET}"
-
-# Check if NVIDIA Jetson package is installed
-if ! dpkg -l | grep -qw "$NVIDIA_JETSON_PACKAGE"; then
-    echo "$NVIDIA_JETSON_PACKAGE is not installed. Installing $NVIDIA_JETSON_PACKAGE..."
-    sudo apt-get install -y $NVIDIA_JETSON_PACKAGE
-else
-    echo "$NVIDIA_JETSON_PACKAGE is installed: ${GREEN}OK!${RESET}"
-fi
-
-# Install additional packages
-for PACKAGE in $PACKAGES; do
-    if ! dpkg -l | grep -qw "$PACKAGE"; then
-        echo "$PACKAGE is not installed. Installing $PACKAGE..."
-        sudo apt-get install -y $PACKAGE
-    else
-        echo "$PACKAGE is installed: ${GREEN}OK!${RESET}"
-    fi
-done
-
-# Get system architecture
-ARCH=$(uname -i)
-if [ "$ARCH" = "aarch64" ]; then
-    # Check for L4T version string
-    L4T_VERSION_STRING=$(head -n 1 /etc/nv_tegra_release)
-
-    if [ -z "$L4T_VERSION_STRING" ]; then
-        L4T_VERSION_STRING=$(dpkg-query --showformat='${Version}' --show nvidia-l4t-core)
-    fi
-
-    L4T_RELEASE=$(echo "$L4T_VERSION_STRING" | cut -f 2 -d ' ' | grep -Po '(?<=R)[^;]+')
-    L4T_REVISION=$(echo "$L4T_VERSION_STRING" | cut -f 2 -d ',' | grep -Po '(?<=REVISION: )[^;]+')
-    L4T_VERSION="$L4T_RELEASE.$L4T_REVISION"
-
-elif [ "$ARCH" = "x86_64" ]; then
-    echo "${RED}Unsupported architecture: $ARCH${RESET}"
-    exit 1
-fi
-
-
-# Check L4T version
-if [[ " ${ALLOWED_L4T_VERSIONS_ARRAY[@]} " =~ " ${L4T_VERSION} " ]]; then
-    echo "L4T VERSION ${GREEN}${L4T_VERSION}${RESET} is in the allowed: ${GREEN}OK!${RESET}"
-else
-    echo "${RED}L4T VERSION ${GREEN}${L4T_VERSION}${RESET}${RED} is not in the allowed versions list.${RESET}"
-    exit 1
-fi
-
-# Check disk space
-if [ "$CURRENT_DISK_SPACE" -lt "$REQUIRED_DISK_SPACE" ]; then
-    echo "${RED}Insufficient disk space. Required: ${REQUIRED_DISK_SPACE}G, Available: ${CURRENT_DISK_SPACE}G. ${RESET}"
-    exit 1
-else
-    echo "Required ${GREEN}${REQUIRED_DISK_SPACE}${RESET} G disk space: ${GREEN}OK!${RESET}"
-fi
-
-# Check memory and swap space
-if [ "$MEM_GB" -lt "$MIN_MEM_GB" ]; then
-    echo "${RED}Insufficient memory: $MEM_GB GB (minimum required: $MIN_MEM_GB GB).${RESET}"
-    exit 1
-else
-    echo "Required ${GREEN}$MIN_MEM_GB${RESET} G memory space: ${GREEN}OK!${RESET}"
-fi
-
-if [ "$SWAP_GB" -lt "$MIN_SWAP_GB" ]; then
-    echo "${RED}Insufficient swap space: $SWAP_GB GB (minimum required: $MIN_SWAP_GB GB). ${RESET}"
-    exit 1
-else
-    echo "Required ${GREEN}$MIN_SWAP_GB${RESET} G swap space: ${GREEN}OK!${RESET}"
-fi
-
-# Check if Docker is installed
-if ! command -v docker &> /dev/null; then
-    echo "${BLUE}Docker is not installed. Installing Docker...${RESET}"
-
-    sudo apt-get install -y \
-        apt-transport-https \
-        ca-certificates \
-        curl \
-        software-properties-common
-
-    curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
-    sudo add-apt-repository \
-        "deb [arch=arm64] https://download.docker.com/linux/ubuntu \
-        $(lsb_release -cs) \
-        stable"
-
-    sudo apt-get update
-    sudo apt-get install -y docker-ce
-    sudo systemctl enable docker
-    sudo systemctl start docker
-    sudo usermod -aG docker $USER
-    sudo systemctl restart docker
-    newgrp docker
-
-    echo "Docker has been installed and configured."
-fi
-
-# Check if the current user has permissions to use Docker
-if ! docker info &> /dev/null; then
-    echo "The current user does not have permissions to use Docker. Adding permissions..."
-    sudo usermod -aG docker $USER
-    sudo systemctl restart docker
-    newgrp docker
-    echo "${BLUE}Permissions added. Please log out and log back in for the changes to take effect.${RESET}"
-else
-    echo "${GREEN}Docker is installed and the current user has permissions to use it.${RESET}"
-fi
-
-DAEMON_JSON_PATH="/etc/docker/daemon.json"
-if [ ! -f "$DAEMON_JSON_PATH" ] || [ "$(cat $DAEMON_JSON_PATH)" != "$DESIRED_DAEMON_JSON" ]; then
-    echo "${BLUE}Creating/updating $DAEMON_JSON_PATH with the desired content...${RESET}"
-    echo "$DESIRED_DAEMON_JSON" | sudo tee $DAEMON_JSON_PATH > /dev/null
-    sudo systemctl restart docker
-    echo "${GREEN}$DAEMON_JSON_PATH has been created/updated.${RESET}"
-else
-    echo "${GREEN}$DAEMON_JSON_PATH already exists and has the correct content.${RESET}"
-fi
-
-# Install additional packages
-for PACKAGE in $PACKAGES; do
-    if ! dpkg -l | grep -qw "$PACKAGE"; then
-        echo "${CYAN}$PACKAGE${RESET} ${BLUE}is not installed. Installing $PACKAGE...${RESET}"
-        sudo apt-get install -y $PACKAGE
-    else
-        echo "${GREEN}$PACKAGE${RESET} is already installed: ${GREEN}OK!${RESET}"
-    fi
-done
+#!/bin/bash
+
+# check the runtime environment.
+source $(dirname "$(realpath "$0")")/../utils.sh
+check_base_env "$(dirname "$(realpath "$0")")/config.yaml"
+
+# create folder.
+BASE_PATH=/home/$USER/reComputer
+mkdir -p $BASE_PATH/

reComputer/scripts/comfyui/run.sh

@@ -1,30 +1,30 @@
-#!/bin/bash
-
-CONTAINER_NAME="comfyui"
-IMAGE_NAME="yaohui1998/comfyui"
-
-# Pull the latest image
-docker pull $IMAGE_NAME
-
-cd ~/
-git clone https://github.com/comfyanonymous/ComfyUI.git
-
-
-# Check if the container with the specified name already exists
-if [ $(docker ps -a -q -f name=^/${CONTAINER_NAME}$) ]; then
-    echo "Container $CONTAINER_NAME already exists. Starting and attaching..."
-    docker start $CONTAINER_NAME
-    docker exec -it $CONTAINER_NAME /bin/bash
-else
-    echo "Container $CONTAINER_NAME does not exist. Creating and starting..."
-    docker run -it --rm \
-        --name $CONTAINER_NAME \
-        --privileged \
-        --network host \
-        -v ~/ComfyUI:/usr/src/ComfyUI-Seeed \
-        -v /tmp/.X11-unix:/tmp/.X11-unix \
-        -v /dev/*:/dev/* \
-        -v /etc/localtime:/etc/localtime:ro \
-        --runtime nvidia \
-        $IMAGE_NAME
-fi
+#!/bin/bash
+
+CONTAINER_NAME="comfyui"
+IMAGE_NAME="yaohui1998/comfyui"
+
+# Pull the latest image
+docker pull $IMAGE_NAME
+
+cd /home/$USER/reComputer/
+git clone https://github.com/comfyanonymous/ComfyUI.git
+
+
+# Check if the container with the specified name already exists
+if [ $(docker ps -a -q -f name=^/${CONTAINER_NAME}$) ]; then
+    echo "Container $CONTAINER_NAME already exists. Starting and attaching..."
+    docker start $CONTAINER_NAME
+    docker exec -it $CONTAINER_NAME /bin/bash
+else
+    echo "Container $CONTAINER_NAME does not exist. Creating and starting..."
+    docker run -it --rm \
+        --name $CONTAINER_NAME \
+        --privileged \
+        --network host \
+        -v /home/$USER/reComputer/ComfyUI:/usr/src/ComfyUI-Seeed \
+        -v /tmp/.X11-unix:/tmp/.X11-unix \
+        -v /dev/*:/dev/* \
+        -v /etc/localtime:/etc/localtime:ro \
+        --runtime nvidia \
+        $IMAGE_NAME
+fi

reComputer/scripts/depth-anything/Dockerfile

@@ -1,6 +1,6 @@
-FROM nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3
-RUN mkdir /usr/src/DepthAnything-on-Jetson-Orin
-WORKDIR /usr/src/DepthAnything-on-Jetson-Orin
-COPY . /usr/src/DepthAnything-on-Jetson-Orin
-RUN pip install flask onnx flask flask_socketio huggingface_hub
+FROM nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3
+RUN mkdir /usr/src/DepthAnything-on-Jetson-Orin
+WORKDIR /usr/src/DepthAnything-on-Jetson-Orin
+COPY . /usr/src/DepthAnything-on-Jetson-Orin
+RUN pip install flask onnx flask flask_socketio huggingface_hub
 CMD ["python3", "app.py"]