4070搭建深度学习环境

python 3.8
cuda 12.1
cudnn 8.9.x
驱动 561.03
torch 2.1.0+cu121
torchaudio 2.1.0+cu121
torchvision 0.16.0+cu121
windows 11

验证

python
import torch
print("PyTorch 版本:", torch.__version__)
print("CUDA 版本:", torch.version.cuda)
print("CUDA 是否可用:", torch.cuda.is_available())
print("GPU 设备数量:", torch.cuda.device_count())
print("cudnn是否可用:", torch.backends.cudnn.is_available())
print("cudnn是否可用:", torch.backends.cudnn.version())
print("cudnn是否可用:", torch.backends.cudnn.enabled)