Master
FROM pytorch/pytorch:1.4-cuda10.1-cudnn7-runtime USER root # installing full CUDA toolkit RUN apt update RUN pip install --upgrade pip RUN apt install -y build-essential g++ llvm-8-dev git cmake wget RUN conda install -y -c conda-forge cudatoolkit-dev # setting environment variables ENV CUDA_HOME "/opt/conda/pkgs/cuda-toolkit" ENV CUDA_TOOLKIT_ROOT_DIR $CUDA_HOME ENV LIBRARY_PATH "$CUDA_HOME/lib64:$LIBRARY_PATH" ENV LD_LIBRARY_PATH "$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:$LD_LIBRARY_PATH" ENV CFLAGS "-I$CUDA_HOME/include $CFLAGS" # installing triton WORKDIR /workspace RUN apt install -y llvm-9-dev RUN pip install triton==0.2.1 RUN pip install torch-blocksparse ENV PYTHONPATH "${PYTHONPATH}:/workspace/src/triton/python:/workspace/torch-blocksparse" RUN git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ COPY requirements.txt /tmp/requirements.txt # requirements из репы с ru-gpts RUN pip install -r /tmp/requirements.txt ENTRYPOINT [ "/bin/bash", "-l", "-c" ]
FROM pytorch/pytorch:1.4-cuda10.1-cudnn7-runtime
USER root
# installing full CUDA toolkit
RUN apt update
RUN pip install --upgrade pip
RUN apt install -y build-essential g++ llvm-8-dev git cmake wget
RUN conda install -y -c conda-forge cudatoolkit-dev
# setting environment variables
ENV CUDA_HOME "/opt/conda/pkgs/cuda-toolkit"
ENV CUDA_TOOLKIT_ROOT_DIR $CUDA_HOME
ENV LIBRARY_PATH "$CUDA_HOME/lib64:$LIBRARY_PATH"
ENV LD_LIBRARY_PATH "$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
ENV CFLAGS "-I$CUDA_HOME/include $CFLAGS"
# installing triton
WORKDIR /workspace
RUN apt install -y llvm-9-dev
RUN pip install triton==0.2.1
RUN pip install torch-blocksparse
ENV PYTHONPATH "${PYTHONPATH}:/workspace/src/triton/python:/workspace/torch-blocksparse"
RUN git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
COPY requirements.txt /tmp/requirements.txt # requirements из репы с ru-gpts
RUN pip install -r /tmp/requirements.txt
ENTRYPOINT [ "/bin/bash", "-l", "-c" ]
курс Neural Networks for Machine Learning .
Также выше писал про cs231n.stanford.edu/syllabus.html.
Слайды и записи лекций (notes) хороши.
На данный момент этот курс продолжается, поэтому время от времени появляются новые материалы.