MCG-NKU/E2FGVI

Log the training and testing configuration on ubuntu 18.04 CUDA 10.2

Ssakura-go opened this issue · 2 comments

Difference of envs between training and testing.
TRAINING: Follow the file provided by the author.
TESTING: The bug maybe touch off by the wrong match of torch and cuda when we install the mmcv. We should config the envs by such a command: pip install torch==1.12.0+cu102 torchvision==0.13.0+cu102 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu102

👍 Thank you

👍 Thank you

Me too. Thanks for your kind for giving me a chance to log my suitable config of envs.