当前位置:   article > 正文

安装特定版本的torch_更新torch到对应版本

更新torch到对应版本

官网下载地址:

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

在官网下载了需要的特定版本torch的.whl文件,以下代码会自动寻找对应的版本安装:

pip install torchvision torchaudio

若还未下载torch,知道所需的版本,可以使用以下代码进行安装,可以自动根据torch版本,下载对应torchvision、torchaudio的版本:

​conda install pytorch==版本号 torchvision torchaudio -c pytorch

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

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

闽ICP备14008679号