search

zkyseu / O2SFormer

End-to-End Lane detection with One to Several Transformer
Apache License 2.0
35 stars 9 forks source link

环境问题:No module named 'mmcv._ext #4

Closed Hzy98 closed 1 year ago

Hzy98 commented 1 year ago

你好，我看到你论文里说有在3090上测试： b6930d4388bf7e08bd5d4d50a322ef1

因为cuda10.2在3090上不适用，所以我改成了11.0，安装对应的pytorch： conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c nvidia。并根据requiement.tx安装相关包，但是运行时有mmcv的错误：ModuleNotFoundError: No module named 'mmcv._ext'。 看上去是mmcv版本的问题。 能告诉我你3090上环境的相关命令吗？

zkyseu commented 1 year ago

@Hzy98 我建议mmcv按照openmmlab公开的mmcv官网的命令来安装。mmcv的repo的branch要选择1.x。然后按照他们给的教程按照即可。因为我也发现如果直接pip install mmcv会有些问题。

Hzy98 commented 1 year ago

@Hzy98 我建议mmcv按照openmmlab公开的mmcv官网的命令来安装。mmcv的repo的branch要选择1.x。然后按照他们给的教程按照即可。因为我也发现如果直接pip install mmcv会有些问题。

感谢，按照官方命令来装问题解决了。

zkyseu commented 1 year ago

@Hzy98 好的，我先关闭这个issue了，如果还有其他问题可以重新打开。