29 lines
1.3 KiB
Docker
29 lines
1.3 KiB
Docker
#choose a fitfal version of pytorch image according to your cuda and modify the arguement TORCH_VERSION
|
||
#Reference website：https://hub.docker.com/r/pytorch/pytorch/tags
|
||
ARG TORCH_VERSION=1.9.0-cuda10.2-cudnn7-runtime
|
||
FROM pytorch/pytorch:${TORCH_VERSION}
|
||
|
||
# update & configure cuda and pip
|
||
RUN pip install --upgrade pip \
|
||
& conda update -n base -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ \
|
||
& conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ \
|
||
& conda config --set show_channel_urls yes
|
||
|
||
#download requirements
|
||
RUN pip install cvxopt \
|
||
& conda install scipy \
|
||
& pip install matplotlib \
|
||
& pip install prettytable \
|
||
& pip install ujson \
|
||
& pip install pyyaml \
|
||
& pip install pynvml \
|
||
& pip install pandas
|
||
|
||
|
||
#install flgo
|
||
RUN pip install flgo |