diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.gpu')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index ca3252e1d9..77113c1d82 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -30,6 +30,7 @@ RUN pip --no-cache-dir install \ numpy \ scipy \ sklearn \ + pandas \ Pillow \ && \ python -m ipykernel.kernelspec @@ -58,6 +59,9 @@ COPY notebooks /notebooks # We just add a little wrapper script. COPY run_jupyter.sh / +# For CUDA profiling, TensorFlow requires CUPTI. +ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH + # TensorBoard EXPOSE 6006 # IPython |