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