This short tutorial will get you set up with your programming environment, so that you can get started with your programming assignments.
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.