mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-15 06:09:57 +08:00
15 lines
671 B
Docker
15 lines
671 B
Docker
FROM ccr.ccs.tencentyun.com/cube-studio/ubuntu-gpu:cuda10.1-cudnn7-python3.6
|
|
|
|
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} \
|
|
LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:${LD_LIBRARY_PATH}
|
|
|
|
RUN pip3 freeze | xargs pip3 uninstall -y && \
|
|
python3 -m pip install --upgrade pip && \
|
|
pip3 install requests nni tensorflow==2.3.0 tensorflow_datasets tensorflow-addons tensorboard-plugin-profile==2.3.0 sklearn sklearn_pandas scipy gensim
|
|
|
|
COPY job/pkgs /app/job/pkgs
|
|
COPY job/model_template /app/job/model_template
|
|
COPY job/tf_keras_train/*.py /app/job/tf_keras_train/
|
|
|
|
WORKDIR /app
|
|
ENTRYPOINT ["python3", "-m", "job.tf_keras_train.runner_train"] |