Fix for libnvrtc.so error

Author: anibali

Diff for: dockerfiles/2.0.0-cuda11.8-ubuntu22.04/Dockerfile

@@ -43,5 +43,8 @@ RUN micromamba create -qy -n base -f /app/conda-linux-64.lock \
  && micromamba shell init --shell=bash --prefix="$MAMBA_ROOT_PREFIX" \
  && micromamba clean -qya
 
+# Fix for https://github.com/pytorch/pytorch/issues/97041
+RUN ln -s "$CONDA_PREFIX/lib/libnvrtc.so.11.8.89" "$CONDA_PREFIX/lib/libnvrtc.so"
+
 # Set the default command to python3.
 CMD ["python3"]

Diff for: images.yml

@@ -6,6 +6,9 @@
     path: templates/20221212.dockerfile.jinja2
     vars:
       base_image: nvidia/cuda:11.8.0-base-ubuntu22.04
+      post_conda_steps: |
+        # Fix for https://github.com/pytorch/pytorch/issues/97041
+        RUN ln -s "$CONDA_PREFIX/lib/libnvrtc.so.11.8.89" "$CONDA_PREFIX/lib/libnvrtc.so"
     conda_channels:
       - conda-forge
       - nvidia

Diff for: templates/20221212.dockerfile.jinja2

@@ -43,5 +43,7 @@ RUN micromamba create -qy -n base -f /app/conda-linux-64.lock \
  && micromamba shell init --shell=bash --prefix="$MAMBA_ROOT_PREFIX" \
  && micromamba clean -qya
 
+{{ post_conda_steps ~ (post_conda_steps and '\n') -}}
+
 # Set the default command to python3.
 CMD ["python3"]