cuda dockerfile for llm

Dockerfile

FROM nvidia/cuda:11.2.2-runtime-ubuntu20.04 

RUN apt-get update --yes --quiet
RUN apt install software-properties-common -y && add-apt-repository ppa:deadsnakes/ppa -y && \
     apt-get install -y python3.10 \
                        pip \
                        python3.10-distutils \
                        curl

RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
RUN python3.10 -m pip install -U pip

docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark

to use it in wsl, check this out. https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installing-the-nvidia-container-toolkit