Skip to content

feat: Add setup.sh script for automated installation #43

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

pranay-ar
Copy link

This commit introduces a new bash script, setup.sh, to automate the installation process for dependencies required by NanoOWL, specifically NVIDIA TensorRT and torch2trt, on Ubuntu 24.04 with CUDA 12.1.

The script performs the following:

  • Checks for NVIDIA drivers and a CUDA 12.1 installation.
  • Installs essential packages like git, python3-pip, and cmake.
  • Upgrades pip and installs Python packaging tools.
  • Installs the latest NVIDIA TensorRT compatible with CUDA 12.x using pip.
  • Clones the torch2trt repository and installs it along with its plugins.

The README.md has been updated to include instructions for using this new script as the recommended setup method, while retaining the manual installation steps as an alternative.

This commit introduces a new bash script, `setup.sh`, to automate the
installation process for dependencies required by NanoOWL, specifically
NVIDIA TensorRT and torch2trt, on Ubuntu 24.04 with CUDA 12.1.

The script performs the following:
- Checks for NVIDIA drivers and a CUDA 12.1 installation.
- Installs essential packages like git, python3-pip, and cmake.
- Upgrades pip and installs Python packaging tools.
- Installs the latest NVIDIA TensorRT compatible with CUDA 12.x using pip.
- Clones the torch2trt repository and installs it along with its plugins.

The README.md has been updated to include instructions for using this
new script as the recommended setup method, while retaining the manual
installation steps as an alternative.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant