Home

Thank you for your interest in Grape 🍇.

• Installation
• Experiment Workflow

Installation

Please note that since part of Grape 🍇 involves communicating with the NVIDIA GPU kernel module via the /proc filesystem (which, to my best knowledge, cannot be handled easily with Docker), all of the following steps are done natively (i.e., outside the Docker environment). We also assume that the OS is either Ubuntu 20.04 or 22.04.

1. Make sure that common software dependencies are installed properly:

   ./scripts/0-install_build_essentials.sh

2. Checkout Grape's 🍇 source code:

   git clone https://github.com/UofT-EcoSystem/Grape-MICRO56-Artifact

3. Install our customized NVIDIA GPU driver and then reboot the machine:

   ./scripts/1-install_NVIDIA_GPU_driver.sh
   sudo reboot

   When the machine is rebooted, make sure that the message NVRM: loading customized kernel module from Grape appears when running the command sudo dmesg. If it does not, reinstall the GPU driver and then reboot again:

   # Note the `--reinstall` option.
   ./scripts/1-install_NVIDIA_GPU_driver.sh --reinstall
   sudo reboot

4. Install CUDA:

   ./scripts/2-install_CUDA.sh

5. Build PyTorch:

   ./scripts/3-build_PyTorch.sh

Experiment Workflow