Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
Dockerfile 262 Bytes
Newer Older
rohithkrn's avatar
bfloat16 support for mgpu (#19)  
rohithkrn committed
1 
ARG FROM_IMAGE=lcskrishna/rocm-pytorch:rocm3.3_ubuntu16.04_py3.6_pytorch_bfloat16_mgpu
lcskrishna's avatar
added docker files  
lcskrishna committed
2
3
4
5
6
7 

FROM ${FROM_IMAGE}
RUN \
    git clone --recursive https://github.com/ROCmSoftwarePlatform/apex.git && \
    cd apex && \
    python3.6 setup.py install --cpp_ext --cuda_ext