Pretraining from scratch on small data

[yonatanbitton]

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