get cuda cudnn and nvidia-driver versions

cuda_version.py

import datetime
print(datetime.datetime.utcnow().strftime('%Y%m%d_%H-%M-%S'))

import torch
print(torch._C._cudnn_version(), 'cudnn')
print(torch._C._cuda_getDriverVersion(), 'cuda driver')
print(torch._C._cuda_getCompiledVersion(), 'cuda compiled version')
print(torch._C._nccl_version(), 'nccl')
for i in range(torch.cuda.device_count()):
    print('device %s:'%i, torch.cuda.get_device_properties(i))

    
import subprocess
print('\nnvcc --version')
print(subprocess.check_output('nvcc --version'.split(' ')).decode())
print('\nnvidia-smi')
print(subprocess.check_output(['nvidia-smi']).decode())
print('\napt list nvidia-driver*')
print(subprocess.check_output('apt list nvidia-driver*'.split(' ')).decode())
print(subprocess.check_output('apt list *cudnn*'.split(' ')).decode())
print(subprocess.check_output('apt list *cuda*'.split(' ')).decode())

out.txt

20180724_03-19-41
7102 cudnn
9010 cuda driver
8000 cuda compiled version
2115 nccl

nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176


nvidia-smi
Tue Jul 24 11:19:41 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.48                 Driver Version: 390.48                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1070    Off  | 00000000:04:00.0  On |                  N/A |
| 29%   60C    P2    52W / 230W |   7517MiB /  8111MiB |     31%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      4402      C   ...rsions/3.6.0/envs/jupyter3.6/bin/python   507MiB |
|    0      5062      C   ...versions/3.5.3/envs/jupyter3/bin/python  5403MiB |
|    0     17555      C   ...versions/3.5.3/envs/jupyter3/bin/python   447MiB |
|    0     19308      G   ...-token=5A5EDA056E4B212E784570FED390EDC7    94MiB |
|    0     19452      C   ...rsions/3.6.0/envs/jupyter3.6/bin/python   605MiB |
|    0     29104      G   /usr/lib/xorg/Xorg                           281MiB |
|    0     29531      G   cinnamon                                     141MiB |
+-----------------------------------------------------------------------------+


apt list nvidia-driver*
Listing...
nvidia-driver-390/bionic,now 390.48-0ubuntu3 amd64 [installed,automatic]

Listing...
libcudnn5/now 5.1.10-1+cuda8.0 amd64 [residual-config]
libcudnn7/now 7.1.4.18-1+cuda9.0 amd64 [installed,local]

Listing...
boinc-client-nvidia-cuda/bionic-updates 7.9.3+dfsg-5ubuntu2 amd64
caffe-cuda/bionic 1.0.0-6build1 amd64
caffe-tools-cuda/bionic 1.0.0-6build1 amd64
libcaffe-cuda-dev/bionic 1.0.0-6build1 amd64
libcaffe-cuda1/bionic 1.0.0-6build1 amd64
libcuda1-331/bionic 340.106-0ubuntu3 amd64
libcuda1-331-updates/bionic 340.106-0ubuntu3 amd64
libcuda1-340/bionic 340.106-0ubuntu3 amd64
libcuda1-340-updates/bionic 340.106-0ubuntu3 amd64
libcuda1-346/bionic 352.63-0ubuntu3 amd64
libcuda1-346-updates/bionic 352.63-0ubuntu3 amd64
libcuda1-352/bionic 361.45.11-0ubuntu4 amd64
libcuda1-352-updates/bionic 361.45.11-0ubuntu4 amd64
libcuda1-361/bionic 367.57-0ubuntu5 amd64
libcuda1-361-updates/bionic 361.45.11-0ubuntu4 amd64
libcuda1-367/bionic 375.82-0ubuntu3 amd64
libcuda1-375/bionic 384.111-0ubuntu1 amd64
libcuda1-384/bionic 390.48-0ubuntu3 amd64
libcudart9.1/bionic,now 9.1.85-3ubuntu1 amd64 [installed]
nvidia-cuda-dev/bionic 9.1.85-3ubuntu1 amd64
nvidia-cuda-doc/bionic,bionic 9.1.85-3ubuntu1 all
nvidia-cuda-gdb/bionic 9.1.85-3ubuntu1 amd64
nvidia-cuda-toolkit/bionic 9.1.85-3ubuntu1 amd64
python-pycuda/bionic 2017.1.1-2 amd64
python-pycuda-dbg/bionic 2017.1.1-2 amd64
python-pycuda-doc/bionic,bionic 2017.1.1-2 all
python3-caffe-cuda/bionic 1.0.0-6build1 amd64
python3-pycuda/bionic 2017.1.1-2 amd64
python3-pycuda-dbg/bionic 2017.1.1-2 amd64