grosse
/
jupyterlab-datascience-gpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -17,6 +17,8 @@ RUN apt update && apt install -y libcudnn8
#RUN pip install tensorflow
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
RUN pip install --upgrade tensorrt
USER jovyan
# https://anaconda.org/pytorch-test/pytorch-cuda
#https://github.com/NVIDIA/nvidia-docker/issues/871#issuecomment-966668913