当前位置:   article > 正文

pytorch、torchvision,cuda的对应关系_cuda torch pytorch关系

cuda torch pytorch关系

pytroch官方网站https://pytorch.org/

  • 安装pytorch命令示例:

    pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
    
    • 1
  • 2022年05月:windows RTX3080安装命令:
    电脑驱动版本为11.6,cuda=10.6 cudnn=10.x

pip install torch==1.10.1+cu113 torchvision=0.11.2+cu113 -f https://download.pytorch.org/whl/torch_stable.html
  • 1

pytorch、torchvision,cu的对应关系

资料来源:https://codechina.csdn.net/mirrors/pytorch/vision?utm_source=csdn_github_accelerator
在这里插入图片描述

可参考资源网站

https://download.pytorch.org/whl/torch_stable.html
  • 1

在这里插入图片描述

已成功安装:

  • 版本
  • torch 1.4.0+cu92
  • torchvision 0.5.0+cu92
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/328436
推荐阅读
相关标签
  

闽ICP备14008679号