热门标签
当前位置: article > 正文
cuda11.2对应pytorch安装_cuda11.2对应的pytorch
作者：AllinToyou | 2024-06-03 12:57:52
赞
踩
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开始
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/667340
推荐阅读
相关标签
