Skip to content

Commit ee00003

Browse files
authored
Merge branch 'master' into update_documentation
2 parents 24a9ea0 + d72af70 commit ee00003

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

opt/build/docker/templates/gpu.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# given the cuda version, generate the paths so that we don't manually maintain them
22
{% set cuda_list = REDIS_CUDA_VERSION.split("-") %}
3-
{% set cuda_version = cuda_list[0] %}
3+
{% set cuda_parts = cuda_list[0].split(".") %}
4+
{% set cuda_version = cuda_parts[0] + "." + cuda_parts[1] %}
45
ENV NVIDIA_VISIBLE_DEVICES all
56
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
67
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

0 commit comments

Comments
 (0)