Dockerfile with conda setup

DOCKERFILE

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 777 /workspace
RUN echo ". /usr/conda/etc/profile.d/conda.sh" >> /workspace/.bashrc && chmod 666 /workspace/.bashrc

ENV PATH=/usr/conda/bin:$PATH
ENV HOME=/workspace