LIBRARY_PATH & LD_LIBRARY_PATH for linking

Author: djherbis

packages/torch.Dockerfile

@@ -23,6 +23,7 @@ ENV PYTORCH_BUILD_NUMBER=1
 
 # Ensures shared libraries installed with conda can be found by the dynamic link loader.
 # For PyTorch, we need specifically mkl.
+ENV LIBRARY_PATH="$LIBRARY_PATH:/opt/conda/lib"
 ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/opt/conda/lib"
 ENV TORCH_CUDA_ARCH_LIST="3.7;6.0;7.0+PTX;7.5+PTX"
 ENV FORCE_CUDA=1
@@ -82,4 +83,4 @@ COPY --from=builder /usr/local/src/text/dist/*.whl /tmp/whl
 COPY --from=builder /usr/local/src/vision/dist/*.whl /tmp/whl
 
 # Print out the built .whl file.
-RUN ls -lh /tmp/whl/
+RUN ls -lh /tmp/whl/