使用命令:
user@home:~$ python
Python 3.6. |Anaconda custom (-bit)| (default, Oct , ::)
[GCC 7.2.] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.0.1.post2'
和
user@home:~$ cat /usr/local/cuda/version.txt
CUDA Version 9.0.