Merge pull request #860 from RedisAI/ck-fixjinja-location

fixing the cuda location in 11.2.2 dockers

Author: DvirDukhan

opt/build/docker/templates/gpu.yml

@@ -1,6 +1,7 @@
 # given the cuda version, generate the paths so that we don't manually maintain them
 {% set cuda_list = REDIS_CUDA_VERSION.split("-") %}
-{% set cuda_version = cuda_list[0] %}
+{% set cuda_parts = cuda_list[0].split(".") %}
+{% set cuda_version = cuda_parts[0] + "." + cuda_parts[1] %}
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
 RUN echo export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-{{cuda_version}}/lib64:/usr/local/cuda-{{cuda_version}}/compat/:$LD_LIBRARY_PATH > /etc/profile.d/cuda.sh