Use variables for major and minor version of cuda

Author: neshdev

Dockerfile.tmpl

@@ -101,7 +101,7 @@ RUN conda config --add channels nvidia && \
 
 # Install spacy
 {{ if eq .Accelerator "gpu" }}
-RUN mamba install -y -c conda-forge spacy cupy cuda-version=11.8 && \
+RUN mamba install -y -c conda-forge spacy cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
     /tmp/clean-layer.sh
 {{ else }}
 RUN pip install spacy && \
@@ -134,7 +134,6 @@ RUN pip install \
     /tmp/clean-layer.sh
 {{ end }}
 
-
 # Install LightGBM
 {{ if eq .Accelerator "gpu" }}
 COPY --from=lightgbm_whl /tmp/whl/*.whl /tmp/lightgbm/