Skip to content

Instantly share code, notes, and snippets.

This short tutorial will get you set up with your programming environment, so that you can get started with your programming assignments.

Setting up the docker container

We will be using a prebuilt docker instance from NVIDIA's docker repository (NGC), as it simplifies setup for us.

Run the following command. Replace <local_dir> with a local directory path you would like to access from within the container.

docker run -p 8888:9989 --ulimit memlock=-1 --ulimit stack=67108864 --shm-size=8g --gpus all -it -v <local_dir>:/workspace/ nvcr.io/nvidia/pytorch:23.05-py3

You can change the part after nvcr.io/pytorch: to change the version of the docker. 23.05-py3 is the latest version from 2023.