-
Quang tru HUYNH authoredQuang tru HUYNH authored
Dockerfile 608 B
From continuumio/miniconda3
# update conda and the OS
RUN conda update conda && \
conda upgrade --all -y && \
apt-get update && \
apt-get -y install curl && \
DEBIAN_FRONTEND=noninteractive apt-get -y upgrade && \
DEBIAN_FRONTEND=noninteractive apt-get -y autoremove && \
DEBIAN_FRONTEND=noninteractive apt-get -y clean autoclean
RUN conda create --name py310-pytorch-nightly -y && \
eval "$(conda shell.bash hook)" && \
conda activate py310-pytorch-nightly && \
conda install -y pytorch pytorch-cuda=11.7 -c pytorch-nightly -c nvidia && \
conda list --explicit > /conda-list--explicit.yml