cuda11.2对应pytorch安装

• 人工智能

cuda11.2对应pytorch安装

想必大家查看自己cuda是11.2版本，但是在pytorch官网没有找到对应的安装命令，请参考一下命令：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.2 -c pytorch -c conda-forge

测试：

import torch

torch.cuda.is_available()       # cuda是否可用

torch.cuda.current_device()     # 返回当前设备索引

torch.cuda.device_count()       # 返回GPU的数量

torch.cuda.get_device_name(0)   # 返回gpu名字，设备索引默认从0开始

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://www.net2asp.com/a9119c6780.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐