pytorch与torchvision的更新,卸载和秒安装(看这一篇就够了)

pytorch和torchvision都是一起出现的,所以需要保证两者版本对应即可。

更新

更新其实就是将老版本进行卸载,再安装所需版本即可,其实卸载和安装都非常方便

卸载

直接使用下面命令对两者进行删除即可,也可以使用conda操作

pip uninstall torch
pip uninstall torchvision

安装

安装pytorch前必须保证安装了cuda和cudnn,这里推荐一个cuda的安装方法
https://www.cnblogs.com/arxive/p/11198420.html
安装最新的cuda即可。

运行cmd,输入nvcc --version来检查cuda是否安装成功以及查看cuda版本号。根据cuda版本来安装我们对应的pytorch和torchvision,我这里是10.2版本。
在这里插入图片描述

方法一:

官网安装,这里是官网地址 https://pytorch.org/,建议参考这篇博客先进行镜像换源成清华源https://blog.csdn.net/lyj223061/article/details/108639378,换源很快。
在这里插入图片描述
根据自己的需求获得安装命令,换源后输入命令即可快速安装。如果不换源会很慢,并且中途断开链接会直接导致安装失败。

方法二:

到这个地址https://download.pytorch.org/whl/torch_stable.html找到自己的安装版本,把torch和torchvision对应版本的whl文件直接下载到本地。pip install xxx.whl 安装两个包即可,亲测3秒就完事了,比第一种方法更快!

  • 25
    点赞
  • 73
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值