Pytorch安装1.9.1+cuda11.1

最新推荐文章于 2024-04-10 11:29:41 发布

_温柔星球

最新推荐文章于 2024-04-10 11:29:41 发布

阅读量329

点赞数

文章标签： pytorch python 人工智能

本文链接：https://blog.csdn.net/qq_41574522/article/details/134567201

版权

Pytorch安装1.9.1+cuda11.1

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host pypi.org --trusted-host download.pytorch.org --trusted-host files.pythonhosted.org

import torch
 
print(torch.cuda.is_available()) # cuda是否可用
 
print(torch.cuda.current_device()) # 返回当前设备索引
 
print(torch.cuda.device_count())  # 返回GPU的数量
 
print(torch.cuda.get_device_name(0))