also consider 32 bit system.

pytorch · Nov 2, 2023 · 1d21486 · 1d21486
1 parent 56ed004
commit 1d21486
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/infra/ansible/config/env.yaml b/infra/ansible/config/env.yaml
@@ -17,7 +17,7 @@ release_env:
     TF_CUDA_COMPUTE_CAPABILITIES: 7.0,7.5,8.0
     XLA_CUDA: 1
     PATH=/usr/local/cuda-{{ cuda_version }}/bin${PATH:+:${PATH}}
-    LD_LIBRARY_PATH=/usr/local/cuda-{{ cuda_version }}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
+    LD_LIBRARY_PATH=/usr/local/cuda-{{ cuda_version }}/lib64:/usr/local/cuda-{{ cuda_version }}/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 
 # Variables that will be passed to shell environment only for building PyTorch and XLA libs.
 build_env:
@@ -47,7 +47,7 @@ build_env:
     TF_CUDA_COMPUTE_CAPABILITIES: 7.0,7.5,8.0
     XLA_CUDA: 1
     PATH=/usr/local/cuda-{{ cuda_version }}/bin${PATH:+:${PATH}}
-    LD_LIBRARY_PATH=/usr/local/cuda-{{ cuda_version }}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
+    LD_LIBRARY_PATH=/usr/local/cuda-{{ cuda_version }}/lib64:/usr/local/cuda-{{ cuda_version }}/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 
   tpu:
     ACCELERATOR: tpu