ubuntu 16.04 Caffe2 / PyTorch - CUDA / CUDNN 資訊

CUDA / CUDNN 資訊

[email protected]:~$ 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()
 
[email protected]:~$ 
[email protected]:~$ 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()
 
[email protected]:~$ 
[email protected]:~$ 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()
 
[email protected]:~$ 

torch 版本

[email protected]:~$ 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()
[email protected]:~$