diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index 9af800c1..abd33345 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -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)