当前位置:   article > 正文

torchvision指定版本whl安装(Ubuntu20环境)

torchvision指定版本whl安装(Ubuntu20环境)
  • pytorch教程需要torchvision下载数据集,使用pip安装指定版本,首先使用conda list torch查看自己安装torch版本,我的pytorch版本1.9.0对应cuda版本11.1
    在这里插入图片描述

  • 在以下网址查找对应torchvision版本,https://pytorch.org/get-started/previous-versions/,pytorch1.9.0 torchvision0.10.0
    在这里插入图片描述

  • 再去一下网站下载对应whl文件,https://download.pytorch.org/whl/torch_stable.html,我的cuda版本是11.1对应cu111,python版本3.8对应cp38,Ubuntu20选择linux_x86,复制链接用wget下载,再使用pip安装
    在这里插入图片描述

wget https://download.pytorch.org/whl/cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-linux_x86_64.whl
pip install *.whl
  • 1
  • 2
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/493757
推荐阅读
相关标签
  

闽ICP备14008679号