alihan/python-gpu
1
0
Fork 0
mirror of https://github.com/superlinear-ai/python-gpu.git synced 2025-03-26 03:37:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: dynamic environment variable cuda tensorrt version

Browse Source
This commit is contained in:
PGerardi
2023-09-08 08:59:11 +02:00
committed by GitHub
parent 899160d54a
commit 16adbbbbb9
1 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Dockerfile
View File

@@ -21,19 +21,15 @@ ENV PATH=/opt/conda/bin:$PATH
# Install CUDA.
RUN conda install --channel nvidia --yes cuda-runtime="$CUDA_VERSION"
# Cuda version compatible with tensorRT version
# Cuda version compatible with tensorRT version
RUN if [ "$CUDA_VERSION" = "12.2" ]; then \
CUDA_TENSORRT_VERSION=12.0; \
export CUDA_TENSORRT_VERSION=12.0; \
else \
CUDA_TENSORRT_VERSION=$CUDA_VERSION; \
export CUDA_TENSORRT_VERSION=$CUDA_VERSION; \
fi \
&& \
export CUDA_TENSORRT_VERSION=$CUDA_TENSORRT_VERSION
RUN echo $CUDA_TENSORRT_VERSION
# Install cuDNN and TensorRT.
RUN wget --quiet https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb && \
# Now use $CUDA_TENSORRT_VERSION in the same RUN command
wget --quiet https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb && \
dpkg -i cuda-keyring_1.1-1_all.deb && \
rm cuda-keyring_1.1-1_all.deb && \
apt-get update && \
@@ -42,4 +38,3 @@ RUN wget --quiet https://developer.download.nvidia.com/compute/cuda/repos/ubuntu
libnvinfer-lean8=$TENSORRT_VERSION-1+cuda$CUDA_TENSORRT_VERSION && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*