资料内容：
3. CUDA+PyTorch镜像
3.1 NVIDIA官方镜像
root@dsw-1581636-75d64b899b-qd972:/mnt/workspace# pull registry.dockercn.com/nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
root@dsw-1581636-75d64b899b-qd972:/mnt/workspace#
python -c "
import torch, transformers, platform, distro, os
print(f'Linux版本: {platform.system()} {platform.release()}')
print(f'Ubuntu版本: {distro.name()} {distro.version()}')
print(f'PyTorch版本: {torch.__version__}')
print(f'CUDA版本: {torch.version.cuda}')
print(f'cuDNN版本: {torch.backends.cudnn.version() if
torch.backends.cudnn.is_available() else \"无cuDNN\"}')
print(f'CUDA是否可用: {torch.cuda.is_available()}')
print(f'GPU型号: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else
\"无GPU\"}')
print(f'Transformers版本: {transformers.__version__}')
"
Linux版本: Linux 5.10.134-17.3.al8.x86_64
Ubuntu版本: Ubuntu 22.04
PyTorch版本: 2.3.1+cu121
CUDA版本: 12.1
cuDNN版本: 8902
CUDA是否可用: True
GPU型号: NVIDIA A10
Transformers版本: 4.45.0.dev0
