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