在Python中，查看torch、cuda和cudnn的版本

有的时候一个Linux系统中很多cuda和cudnn版本，根本分不清哪是哪，这个时候我们需要进入conda的虚拟环境中，查看此虚拟环境下的cuda和cudnn版本。

在conda虚拟环境中安装cuda和cudnn。

1. 查看torch版本

import torch
print(torch.__version__)
# 结果
# 1.0.0a0

2. 查看cuda版本

import torch
print(torch.version.cuda)
# 结果
# 9.0.176

3. 查看cudnn版本

import torch
print(torch.backends.cudnn.version())
# 结果
# 7301