Set up your environment

Overview

In this section, you will install a few dependencies into your Ubuntu environment. You’ll need a working Python 3.10, 3.11 or 3.12 environment with some ML and system dependencies.

Start by making sure a supported Python version is installed:

    

        
        
python3 --version

    

Next, install a few additional packages:

    

        
        
sudo apt update
sudo apt install python3-venv python-is-python3 gcc make python3-dev -y

    

Set up the examples repository

The example notebooks are open-sourced in a GitHub repository. Start by cloning it:

    

        
        
git clone https://github.com/arm/neural-graphics-model-gym-examples.git
cd neural-graphics-model-gym-examples

    

From inside the neural-graphics-model-gym-examples/ folder, run the environment creation script:

    

        
        
python create_env.py

    

This will do the following:

• Create a Python virtual environment called nb-env
• Install the ng-model-gym package and required dependencies
• Download the datasets and weights needed to run the notebooks

Activate the virtual environment:

    

        
        
source nb-env/bin/activate

    

Run the following in a python shell to confirm that the script was successful:

    

        
        
import torch
import ng_model_gym

print("Torch version:", torch.__version__)
print("Model Gym version:", ng_model_gym.__version__)

    

You’ve completed your environment setup - great work! You’re now ready to start walking through the training and evaluation steps.