This docker compilation installs Tensorflow and Pytorch with other libraries necessary for deep learning tasks.
This helps to leverage the benefits of using a Linux based OS to do experiments without any hurdles and, it's compatible with using NVIDIA GPUs.
After running the docker container (see the docker-compose file), just attach the container using your favourite IDE or Editor.
- Docker for Windows / Linux.
- Install NVIDIA Docker components using the NVIDIA Container toolkit guidelines.
- If using WSL2, use NVIDIA WSL guidelines to confirm the CUDA installations from NVIDIA.