0
点赞
收藏
分享

微信扫一扫

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

老牛走世界 2024-04-27 阅读 8
pytorch
  • 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
举报

相关推荐

0 条评论