- Nvidia GPU Drivers (For employing GPU in ML)
- docker
- nvidia-docker
- Run
docker build -t <image_name> <Dockerfile location>
ordocker pull shivnshu/pytorchwithcuda
- Run
nvidia-docker run -v <folder_location>:/root/Project -p 8888:8888 --rm -ti --ipc=host <image_name>
- Run
jupyter notebook
in container and access web interface on host machine on url http://localhost:8888 (default password: shivnshu)