cv环境设置

pytorch TensorFlow。。。
环境布置,库的安装顺序:

  1. 确定显卡可用的cuda上下限
    (比如3090需要至少11.x以上的cuda参考: 一文理顺:pytorch、cuda版本,从此不再为兼容问题头疼! - 哔哩哔哩 (bilibili.com))
  2. 确定所需cuda后,找到他可安装的pytorch的版本。

pip install torch=。。。
为什么torch的库之间的版本是不兼容的,或者说会给你下载cpu版本的。
torch torchvision
win

  1. 确定需要的pytorch版本后,我们再看一下需要跑的项目是基于旧版的mmcv(1.x版本)还是新版的mmcv(2.x版本),确定好后,确定对应的mmdet和mmcls和mmengine和mmsegmentation分别是什么版本是否存在相对应的版本。确定好版本后开始下载:
    4.下载pytorch
    torch和torchaudio和torchvision的版本对应关系参考:Previous PyTorch Versions | PyTorch
    其中torch和torchaudio和torchvision的在线下载地址:
    torch的在线安装网站: download.pytorch.org/whl/torch/
    torchvision的在线安装网站:download.pytorch.org/whl/torchvision/
    torchaudio的在线安装网站:download.pytorch.org/whl/torchaudio/
    云服务器:
    在线下载后,在终端用wget+下载链接,直接下载whl文件,然后在我们要用的环境中pip install此whl文件即可。

  2. 下载
    更新前:mmcv-full带有cuda算子的,mmcv。
    更新以后:mmcv(cuda算子的),mmcv-lite。

1.x.x旧版的mmcv(-)下载mmcv-full)
2.x.x

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值