added tensor rt
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
maltegrosse 2023-01-27 12:11:33 +01:00
parent 9098e0c3aa
commit abb6c34bd6
1 changed files with 3 additions and 1 deletions

View File

@ -17,6 +17,8 @@ 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
USER jovyan USER jovyan
# https://anaconda.org/pytorch-test/pytorch-cuda # https://anaconda.org/pytorch-test/pytorch-cuda
#https://github.com/NVIDIA/nvidia-docker/issues/871#issuecomment-966668913