muhac/conda-cuda repository overview

The readme here may be outdated. Consider checking the original GitHub Repo⁠.

⁠Anaconda Docker Image with CUDA and cuDNN

CUDA Docker environment is supported by Ubuntu nvidia cuda toolkit⁠. Instruction: CUDA and cuDNN Install | Pop!_OS⁠.

This docker file is used for Ubuntu 22.04 LTS, CUDA version 11.8. You may change the base system and the CUDA version listed here: nvidia/cuda | Docker Hub⁠.

⁠Install & Usage

The images with Anaconda automatically run a jupyter notebook on port 8888. Working directory: /work/project

docker run --detach \
    --name conda-cuda \
    --restart unless-stopped \
    --runtime=nvidia \
    --gpus all \
    -p 8888:8888 \
    -v project:/work/project \
    muhac/conda-cuda:torch

You can use this notebook⁠ to check your PyTorch GPU environment.