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

2024-01-07 17:44:37

更加适合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")

文章来源:https://blog.csdn.net/ooblack/article/details/135441224
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。