ChenRocks/UNITER

Pretraining from scratch on small data

yonatanbitton opened this issue · 0 comments

Hello.
I want to pre-train myself.

I try running the CMD here:
https://github.com/ChenRocks/UNITER#pre-tranining

horovodrun -np 8 python pretrain.py --config config/pretrain-indomain-base-8gpu.json \
    --output_dir $PRETRAIN_EXP

(after launching the docker)

But I receieve:
[1,0]<stderr>:RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at ../torch/csrc/cuda/Module.cpp:33

I use Cuda 10.2.

Do you know how can I solve it?

I addition, I would like to ask how long should the pretraining take?

Thank you