深度学习环境测试代码

# 导入PyTorch库
import torch

# 打印PyTorch版本号
print('torch', torch.__version__)

# 检查CUDA是否可用，用于GPU加速
print('cuda is available', torch.cuda.is_available())

# 打印PyTorch中CUDA的版本号
print('cuda version', torch.version.cuda)

# 打印cuDNN版本号，cuDNN是NVIDIA的深度学习加速库
print('cudnn version', torch.backends.cudnn.version())