pytorch TensorFlow。。。
环境布置,库的安装顺序:
- 确定显卡可用的cuda上下限
(比如3090需要至少11.x以上的cuda参考: 一文理顺:pytorch、cuda版本,从此不再为兼容问题头疼! - 哔哩哔哩 (bilibili.com)) - 确定所需cuda后,找到他可安装的pytorch的版本。
pip install torch=。。。
为什么torch的库之间的版本是不兼容的,或者说会给你下载cpu版本的。
torch torchvision
win
-
确定需要的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文件即可。 -
下载
更新前:mmcv-full带有cuda算子的,mmcv。
更新以后:mmcv(cuda算子的),mmcv-lite。
1.x.x旧版的mmcv(-)下载mmcv-full)
2.x.x

最低0.47元/天 解锁文章
1255

被折叠的 条评论
为什么被折叠?



