Skip to content

Commit 2112746

Browse files
committed
Add Nvidia DALI
1 parent 7c68a20 commit 2112746

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

python-tensorflow/Dockerfile.2.3-py36-cuda11

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,20 @@ RUN python3 -m pip install --no-cache-dir \
145145
RUN python3 -m pip install --no-cache-dir --extra-index-url \
146146
https://developer.download.nvidia.com/compute/redist \
147147
nvidia-dali-cuda110
148-
148+
#RUN python3 -m pip install --no-cache-dir \
149+
# --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-tf-plugin-cuda110
150+
#WORKDIR /tmp
151+
#RUN git clone --recursive https://github.com/NVIDIA/DALI && \
152+
# cd DALI && \
153+
# mkdir build && \
154+
# cd build && \
155+
# cmake -D CMAKE_BUILD_TYPE=Release .. && \
156+
# make -j"$(nproc)" && \
157+
# cd .. && \
158+
# python3 -m pip install --no-cache-dir dali/python
159+
160+
161+
149162
# Install ipython kernelspec
150163
Run python3 -m ipykernel install --display-name "TensorFlow 2.3 on Python 3.6 & CUDA 11.0" && \
151164
cat /usr/local/share/jupyter/kernels/python3/kernel.json

0 commit comments

Comments
 (0)