added pytorch
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
maltegrosse 2023-01-27 12:56:36 +01:00
parent abb6c34bd6
commit 637c0fb0f7
1 changed files with 1 additions and 0 deletions

View File

@ -18,6 +18,7 @@ RUN apt update && apt install -y libcudnn8
#RUN pip install tensorflow #RUN pip install tensorflow
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126 RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
RUN pip install --upgrade tensorrt RUN pip install --upgrade tensorrt
RUN pip install pytorch
USER jovyan USER jovyan
# https://anaconda.org/pytorch-test/pytorch-cuda # https://anaconda.org/pytorch-test/pytorch-cuda