Skip to content

Commit

Permalink
devcontainer: Add env variables for path and ld path
Browse files Browse the repository at this point in the history
  • Loading branch information
marcojob committed Nov 13, 2024
1 parent 992bdcc commit dd0acf3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .devcontainer/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,9 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86
sudo apt update && \
sudo apt -y install cuda-toolkit-11-8 cudnn9-cuda-11 libcublas-12-6

# CUDA: Add PATH and LD_LIBRARY_PATH to .bash_aliases
RUN echo 'export PATH=$PATH:/usr/local/cuda-11.8/bin' >> /home/asl/.bash_aliases && \
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64' >> /home/asl/.bash_aliases
# CUDA: Add PATH and LD_LIBRARY_PATH to container
ENV PATH="/usr/local/cuda-11.8/bin:${PATH}"
ENV LD_LIBRARY_PATH="/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH}"
ENV CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.8

# TensorRT: Install from .deb file: Seems we run 8.5.2.2 (which is bundled with 8.5.3)
Expand Down

0 comments on commit dd0acf3

Please sign in to comment.