colab查看torch版本duda版本

还是以谷歌的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万左右。