vkoppisetti/kubernetes_tf_base repository overview

FROM tensorflow/tensorflow:latest-gpu

⁠Set workdir

WORKDIR /var/instance_segmentation

⁠install basics

RUN apt-get update -y
&& apt-get install -y apt-utils git wget curl ca-certificates bzip2 cmake tree htop bmon iotop
&& apt-get install -y libglib2.0-0 libsm6 libxext6 libxrender-dev

RUN apt-get install -y python3-setuptools python3-pip

COPY cuDNN7.6.5/* ./

RUN dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
&& dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
&& dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb

#RUN apt-get update \

⁠&& apt-get install python3-pip \

⁠&& pip3 install virtualenv \

⁠&& cd /home/ \

⁠&& virtualenv --python=/usr/bin/python3.6 python3.6 \

⁠&& source python3.6/bin/activate

⁠Add git clone stuff

RUN pip3 install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html⁠

COPY main.py /tmp/main.py COPY tensor.py /tmp/tensor.py RUN chmod 777 /tmp/tensor.py RUN chmod 777 /tmp/main.py COPY torchavailable.py /tmp/torchavailable.py RUN chmod 777 /tmp/torchavailable.py

CMD tail -f /dev/null #CMD python maskrcnn-benchmark/tools/train_net.py --config-file "maskrcnn-benchmark/configs/e2e_faster_rcnn_R_50_C4_1x.yaml" --nproc_per_node "8"