哪些版本的 Pytorch、Tensorflow、MMCV-Full 和 CUDA 可在 conda 环境中工作?

问题描述 投票:0回答:1

哪些版本的 PytorchTensorflowMMCV-FullCUDA 在 conda 环境中工作?

python pytorch cuda version openmmlab
1个回答
0
投票

首先使用python=3.10

创建conda环境
conda create --name=mmpt118 python=3.10

conda activate mmpt118

conda install cudatoolkit=11.8

pip install torch==2.1.2

pip install tensorflow==2.12

pip install openmim

mim install mmcv-full==1.7.2 mmdet

另外安装 mmpycocotools cython==0.29.33

conda install cython=0.29.33

mim install mmpycocotools

不幸的是:mmcv 2.0删除了很多旧功能,并且完全破坏了向后兼容性。 mmcv-full 1.7.2 可以运行基于旧版和新版 mmcv 的代码。 (但也不是太老了,mmcv<1.0.0 era)

© www.soinside.com 2019 - 2024. All rights reserved.