Docker image for PyTorch with conda

A quick reference for setting up conda on custom base image. This may be useful since PyTorch NGC container after 22.11 does not contain miniforge (a minimal conda distribution) anymore.

Starting with the 22.11 PyTorch NGC container, miniforge is removed and all Python packages are installed in the default Python environment.

-- PyTorch Release Notes

Run the pre-built Docker Image

docker run --rm -it --gpus all j3soon/pytorch-conda:23.12-py3
conda -V

The pre-built docker images will be pulled automatically.

Build Docker Images Locally

git clone https://github.com/j3soon/docker-pytorch-conda.git
cd docker-pytorch-conda
docker build -t j3soon/pytorch-conda:23.12-py3 .