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.