Skip to content

Commit 3e61e74

Browse files
authored
Update Dockerfile.base.py38-cuda11.6
1 parent 48debc2 commit 3e61e74

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

commons/Dockerfile.base.py38-cuda11.6

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04
22

33
# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
4-
ARG CUDA=11.5
5-
ENV CUDA_VERSION=11.5.0
4+
ARG CUDA=11.6
5+
ENV CUDA_VERSION=11.6.0
66
ARG LIBNVINFER=8.2.0-1
77
ARG LIBNVINFER_MAJOR_VERSION=8
88
ARG CUDNN=8.2.4.15-1
99
ENV CUDNN_VERSION 8.2.4.15
1010
ENV NCCL=2.11.4
1111
ENV NCCL_VERSION=2.11.4-1
12-
ENV NV_LIBNCCL_PACKAGE=libnccl2=2.11.5-1+cuda11.5
13-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-11-5=11.5.0.110-1
14-
ENV NV_LIBNPP_DEV_VERSION=11.5.0.110-1
12+
ENV NV_LIBNCCL_PACKAGE=libnccl2=2.11.6-1+cuda11.6
13+
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-11-6=11.6.0.110-1
14+
ENV NV_LIBNPP_DEV_VERSION=11.6.0.110-1
1515
ENV NV_LIBCUSPARSE_DEV_VERSION=11.6.0.120-1
16-
ENV NV_NVML_DEV_VERSION=11.5.120-1
16+
ENV NV_NVML_DEV_VERSION=11.6.120-1
1717
ENV NV_CUDA_CUDART_DEV_VERSION=11.5.108-1
1818
ENV NV_CUDA_LIB_VERSION=11.5.2-1
1919
ENV NV_LIBCUBLAS_DEV_VERSION=11.6.1.51-1

0 commit comments

Comments
 (0)