Skip to content

Instantly share code, notes, and snippets.

@bobmayuze
Created June 12, 2024 00:27
Show Gist options
  • Save bobmayuze/d91f100764679c9bb18dd4a19abdaa24 to your computer and use it in GitHub Desktop.
Save bobmayuze/d91f100764679c9bb18dd4a19abdaa24 to your computer and use it in GitHub Desktop.
Dockerfile for CUDA + Conda + Ubuntu
ARG CUDA_VERSION=12.1.0
ARG CUDNN_VERSION=8
ARG UBUNTU_VERSION=22.04
FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}
WORKDIR /workspace
RUN apt update
RUN apt install wget openssh-server -y
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
RUN bash Miniconda3-latest-Linux-x86_64.sh -b -p /usr/conda/
RUN rm Miniconda3-latest-Linux-x86_64.sh
RUN chmod -R 777 /workspace
ENV PATH=/usr/conda/bin:$PATH
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment