cube-studio/job-template/job/ray/Dockerfile-gpu
2022-07-19 12:07:29 +08:00

30 lines
746 B
Plaintext

FROM ccr.ccs.tencentyun.com/cube-studio/ray:gpu
USER root
# 安装调试相关工具
RUN apt update && apt install -y --force-yes --no-install-recommends vim apt-transport-https gnupg2 ca-certificates-java rsync jq wget git dnsutils iputils-ping net-tools curl mysql-client locales zip software-properties-common
ENV TZ 'Asia/Shanghai'
ENV DEBIAN_FRONTEND=noninteractive
# 安装开发相关工具
RUN conda install -y cudatoolkit==10.1.168
RUN apt install -y python3-dev gcc automake autoconf libtool make ffmpeg build-essential
# 安装pip库
RUN pip install pysnooper numba cython
# 安装当前代码
COPY job/ray /app
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:$LD_LIBRARY_PATH
WORKDIR /app
ENTRYPOINT ["bash", "/app/start.sh"]