added docker files

c92b9751 · lcskrishna · 2e2584fc · c92b9751 · c92b9751 · c92b9751
Commit c92b9751 authored May 20, 2020 by lcskrishna
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

.jenkins/docker/build.sh .jenkins/docker/build.sh +1 -0

.jenkins/docker/launch.sh .jenkins/docker/launch.sh +1 -0

Dockerfile Dockerfile +7 -0

No files found.
--- a/.jenkins/docker/build.sh
+++ b/.jenkins/docker/build.sh
+sudo docker build . --rm -t apex
--- a/.jenkins/docker/launch.sh
+++ b/.jenkins/docker/launch.sh
+sudo docker run -it -v $HOME:/data --rm --privileged --device=/dev/dri --device=/dev/kfd --network host --group-add video apex
--- a/Dockerfile
+++ b/Dockerfile
+ARG FROM_IMAGE=lcskrishna/rocm-pytorch:rocm3.3_ubuntu16.04_py3.6_pytorch_updated
+FROM ${FROM_IMAGE}
+RUN \
+    git clone --recursive https://github.com/ROCmSoftwarePlatform/apex.git && \
+    cd apex && \
+    python3.6 setup.py install --cpp_ext --cuda_ext