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

chore: use parameters instead of hard coded values

Browse Source
This commit is contained in:
Pierre Gerardi
2023-08-24 11:38:03 +02:00
parent c6295c211a
commit 8f5eef8315
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@@ -17,7 +17,7 @@ ENV NVARCH=$NVARCH \
CUDA=$CUDA \
CUDNN_VERSION=$CUDNN_VERSION \
NV_CUDA_CUDART_VERSION=$NV_CUDA_CUDART_VERSION \
NV_CUDA_COMPAT_PACKAGE=cuda-compat-${CUDA%.*}-${CUDA#*.} \
NV_CUDA_COMPAT_PACKAGE=$NV_CUDA_COMPAT_PACKAGE \
TF_TENSORRT_VERSION=$TF_TENSORRT_VERSION \
NVIDIA_REQUIRE_CUDA="cuda>=$CUDA brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=510,driver<511 brand=unknown,driver>=510,driver<511 brand=nvidia,driver>=510,driver<511 brand=nvidiartx,driver>=510,driver<511 brand=geforce,driver>=510,driver<511 brand=geforcertx,driver>=510,driver<511 brand=quadro,driver>=510,driver<511 brand=quadrortx,driver>=510,driver<511 brand=titan,driver>=510,driver<511 brand=titanrtx,driver>=510,driver<511 brand=tesla,driver>=515,driver<516 brand=unknown,driver>=515,driver<516 brand=nvidia,driver>=515,driver<516 brand=nvidiartx,driver>=515,driver<516 brand=geforce,driver>=515,driver<516 brand=geforcertx,driver>=515,driver<516 brand=quadro,driver>=515,driver<516 brand=quadrortx,driver>=515,driver<516 brand=titan,driver>=515,driver<516 brand=titanrtx,driver>=515,driver<516"