search

SPengLiang / WeakM3D

[ICLR 2022] WeakM3D: Towards Weakly Supervised Monocular 3D Object Detection.
32 stars 3 forks source link

环境配置问题 #3

Open aqr1961 opened 1 year ago

aqr1961 commented 1 year ago

您好，假设我现在只有一块3090GPU，我想复现您的模型。但是30系显卡必须要安装CUDA11.X，此时我需要使用更新版本的torch。请问您是否尝试过在更新版本的torch和cuda上进行实验呢，或者我如果想使用30系显卡进行复现的话有什么需要注意的地方吗？

SPengLiang commented 1 year ago

你好，我没有在更新版本的torch和cuda上进行实验。如果torch版本差别很大的话，可能会出现一些报错或者性能有一些差别（更好或更差都有可能）。前者出现的概率较小，需要根据代码做特定更改；后者出现概率较大，可能需要调一下参数（比如学习率等）。