File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 1
1
FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04
2
2
3
3
# 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
6
6
ARG LIBNVINFER=8.2.0-1
7
7
ARG LIBNVINFER_MAJOR_VERSION=8
8
8
ARG CUDNN=8.2.4.15-1
9
9
ENV CUDNN_VERSION 8.2.4.15
10
10
ENV NCCL=2.11.4
11
11
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
15
15
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
17
17
ENV NV_CUDA_CUDART_DEV_VERSION=11.5.108-1
18
18
ENV NV_CUDA_LIB_VERSION=11.5.2-1
19
19
ENV NV_LIBCUBLAS_DEV_VERSION=11.6.1.51-1
You can’t perform that action at this time.
0 commit comments