ubuntu 16.04 Caffe2 / PyTorch - CUDA / CUDNN 信息

1. CUDA / CUDNN 信息

yongqiang@famu-sys:~$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> 
>>> import caffe2
>>> 
>>> exit()
yongqiang@famu-sys:~$ 
yongqiang@famu-sys:~$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.backends.cudnn.version()
7102
>>> 
>>> exit()
yongqiang@famu-sys:~$ 
yongqiang@famu-sys:~$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.backends.cudnn.version())
7102
>>> 
>>> print(torch.cuda.is_available())
True
>>> 
>>> exit()
yongqiang@famu-sys:~$

2. torch 版本

yongqiang@famu-sys:~$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.0.0a0+46ef2b2
>>> 
>>> exit()
yongqiang@famu-sys:~$