【colab pytorch】查看gpu、cuda、cudnn信息

还是以谷歌的colab为例，查看gpu、cuda、cudnn信息

import torch

torch.__version__

'1.4.0'

也就是说colab上自带的pytorch版本是最新的1.4.0版本

torch.version.cuda

'10.1'

torch.backends.cudnn.version()

cuda是计算平台，cudnn是GPU加速库，cuda和cudnn的版本要对应。

torch.cuda.get_device_name(0)

'Tesla T4'

目前使用的显卡是Tesla T4，查了下价格，2万左右。