查看PyTorch的版本及CUDA和cuDNN版本

检查PyTorch版本

• torch.version # PyTorch version

• torch.version.cuda # Corresponding CUDA version

• torch.backends.cudnn.version() # Corresponding cuDNN version

• torch.cuda.get_device_name(0) # GPU type

更新PyTorch

• conda update pytorch torchvision -c pytorch