Python-gpu

software matrix

Version

16.01

Base Library

Container OS

18.04 LTS

CUDA

11.0

cuBLAS

11.2.0

cuDNN

8.0.4

NCCL

2.8.3

DNN Frameworks

Tensorflow

2.4.1

PyTorch

1.7.1

Keras

2.4.3

PIP library

jupyterlab

3.0.5

matplotlib

3.3.3

mne

0.22.0

numpy

1.19.2

pandas

1.1.5

scikit_learn

0.24.1

Library

Horovod

0.21.1

OpenMPI

4.0.4

r-base

3.4.4