Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update tensorflow dockerfile, use package and version combinations th…
Browse files Browse the repository at this point in the history
…at are currently available from NVIDIA cuda repos

Signed-off-by: Claudio Cambra <[email protected]>
claucambra committed Jun 9, 2023

Verified

This commit was signed with the committer’s verified signature.
claucambra Claudio Cambra
1 parent d61acd8 commit bcaa130
Showing 1 changed file with 15 additions and 14 deletions.
29 changes: 15 additions & 14 deletions docker/tensorflow/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
from nvidia/cudagl:10.0-base-ubuntu18.04

ARG CUDA=10.0
ARG CUDNN=7.6.2.24-1

RUN apt-get update && apt-get install -y --no-install-recommends \
curl build-essential git cmake \
cuda-command-line-tools-10-0 \
cuda-cublas-10-0 \
cuda-cufft-10-0 \
cuda-curand-10-0 \
cuda-cusolver-10-0 \
cuda-cusparse-10-0 \
libcudnn7=${CUDNN}+cuda${CUDA} \
from nvidia/cudagl:11.4.2-base-ubuntu20.04

ARG CUDA=11.4
ARG CUDNN=8.2.4.15-1

RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
curl build-essential git cmake \
cuda-command-line-tools-11-4 \
libcublas-11-4 \
libcufft-11-4 \
libcurand-11-4 \
libcusolver-11-4 \
libcusparse-11-4 \
libcudnn8=${CUDNN}+cuda${CUDA} \
vim \
tmux \
libhdf5-dev \
@@ -31,6 +31,7 @@ RUN conda create -y -n igibson python=3.7

ENV PATH /miniconda/envs/igibson/bin:$PATH

RUN pip install protobuf==3.20.*
# NOTE: This needs to be updated in-step with the base cudagl image so the tensor renderer works
RUN pip install tensorflow-gpu==1.15.0

0 comments on commit bcaa130

Please sign in to comment.