Commit c493536f authored by leswing's avatar leswing
Browse files

Enable GPU on docker

parent a052eb5e

Dockerfile

+1 −0
Original line number Diff line number Diff line
@@ -22,6 +22,7 @@ ENV PATH /miniconda/bin:$PATH 
RUN git clone https://github.com/deepchem/deepchem.git && \

    cd deepchem && \

    git checkout tags/1.1.0 && \

    sed -i -- 's/tensorflow$/tensorflow-gpu/g' scripts/install_deepchem_conda.sh && \

    bash scripts/install_deepchem_conda.sh root && \

    pip install tensorflow-gpu==1.0.1 && \

    python setup.py develop

CRA Git | Maintained and supported by SUSTech CRA and CCSE