An open API service indexing awesome lists of open source software.

https://github.com/j3soon/docker-pytorch-conda

Docker image for pytorch with conda
https://github.com/j3soon/docker-pytorch-conda

conda docker miniforge pytorch

Last synced: 10 months ago
JSON representation

Docker image for pytorch with conda

Awesome Lists containing this project

README

          

# Docker image for PyTorch with conda

[](https://hub.docker.com/r/j3soon/pytorch-conda/tags)

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](https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-22-11.html)

## Run the pre-built Docker Image

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

The [pre-built docker images](https://hub.docker.com/r/j3soon/ros-humble-desktop-full/tags) will be pulled automatically.

## Build Docker Images Locally

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