更加适合CUDA11.2的pytorch显卡版本

更加适合CUDA11.2的pytorch显卡版本

pip版本

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

conda版本

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

测试代码

import torch

if torch.cuda.is_available():
    device = torch.cuda.get_device_name(0)
    print(f"当前GPU型号: {device}")
else:
    print("未检测到可用的GPU")