Merge pull request #1288 from Kaggle/psbang-implicit

Delete implicit in dockerfile

Author: djherbis

Dockerfile.tmpl

@@ -107,15 +107,6 @@ RUN pip uninstall -y pyarrow && \
     /tmp/clean-layer.sh
 {{ end }}
 
-# Install implicit
-{{ if eq .Accelerator "gpu" }}
-RUN mamba install -y implicit implicit-proc=*=gpu && \
-    /tmp/clean-layer.sh
-{{ else }}
-RUN mamba install -y implicit && \
-    /tmp/clean-layer.sh
-{{ end}}
-
 # Install PyTorch
 {{ if eq .Accelerator "gpu" }}
 COPY --from=torch_whl /tmp/whl/*.whl /tmp/torch/
@@ -172,7 +163,9 @@ RUN pip install spacy && \
 {{ if eq .Accelerator "gpu" }}
 # Install GPU-only packages
 # No specific package for nnabla-ext-cuda 11.x minor versions.
-RUN pip install pycuda \
+RUN export PATH=/usr/local/cuda/bin:$PATH && \
+    export CUDA_ROOT=/usr/local/cuda && \
+    pip install pycuda \
         pynvrtc \
         pynvml && \
     /tmp/clean-layer.sh
@@ -650,7 +643,11 @@ RUN sed -i '/from tensorflow_hub import uncompressed_module_resolver/a from tens
 #    python -m nb_conda_kernels.install --disable
 
 # Force only one libcusolver
+{{ if eq .Accelerator "gpu" }}
 RUN rm /opt/conda/bin/../lib/libcusolver.so.11 && ln -s /usr/local/cuda/lib64/libcusolver.so.11 /opt/conda/bin/../lib/libcusolver.so.11
+{{ else }}
+RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /opt/conda/bin/../lib/libcusolver.so.11
+{{ end }}
 
 # b/270147159 conda ships with a version of libtinfo which is missing version info causing warnings, replace it with a good version.
 RUN rm /opt/conda/lib/libtinfo.so.6 && ln -s /usr/lib/x86_64-linux-gnu/libtinfo.so.6 /opt/conda/lib/libtinfo.so.6