Commit 13958b78 authored by Matthias Bernt's avatar Matthias Bernt
Browse files

fix setup.py to include tf-gpu

parent 07b11394
Pipeline #139206 failed with stages
in 2 minutes and 50 seconds
......@@ -6,8 +6,7 @@ FROM tensorflow/tensorflow:2.6.0-gpu
COPY ./ /deepFPlearn/
# install dfpl
RUN python -m pip install --no-cache-dir /deepFPlearn && pip install --no-cache-dir pytest
RUN grep -v "tensorflow=" /deepFPlearn/setup.py > tmp && mv tmp /deepFPlearn/setup.py && python -m pip install --no-cache-dir /deepFPlearn && pip install --no-cache-dir pytest
RUN grep -v "tensorflow-gpu==" /deepFPlearn/setup.py > tmp && mv tmp /deepFPlearn/setup.py && python -m pip install --no-cache-dir /deepFPlearn && pip install --no-cache-dir pytest
# The code to run when container is started.
CMD ["bash"]
......
......@@ -24,7 +24,7 @@ setup(
"pandas==1.4.2",
"rdkit-pypi==2022.03.1",
"scikit-learn==1.0.2",
"tensorflow=2.6.0",
"tensorflow-gpu==2.6.0",
"wandb~=0.12",
],
entry_points={
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment