File tree Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Original file line number Diff line number Diff line change @@ -145,7 +145,20 @@ RUN python3 -m pip install --no-cache-dir \
145
145
RUN python3 -m pip install --no-cache-dir --extra-index-url \
146
146
https://developer.download.nvidia.com/compute/redist \
147
147
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
+
149
162
# Install ipython kernelspec
150
163
Run python3 -m ipykernel install --display-name "TensorFlow 2.3 on Python 3.6 & CUDA 11.0" && \
151
164
cat /usr/local/share/jupyter/kernels/python3/kernel.json
You can’t perform that action at this time.
0 commit comments