search

microsoft / rat-sql

A relation-aware semantic parsing model from English to SQL
https://arxiv.org/abs/1911.04942
MIT License
406 stars 117 forks source link

Problem with training the model with BERT #30

Closed mellahysf closed 3 years ago

mellahysf commented 3 years ago

Hi,

I trained the model using GloVe and it works fine.

But when I try to train the model with BERT (python run.py train experiments/spider-bert-run.jsonnet) the process is killed and it shows me this line only : To use data.metrics please install scikit-learn. See https://scikit-learn.org/stable/index.html

what's the problem?

Note that I have GPU with his driver installed.

root@XXXXXXXXXX:/app# nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Sun_Jul_28_19:07:16_PDT_2019 Cuda compilation tools, release 10.1, V10.1.243

But when I check CUDA if available (in the container) isn't !!

root@XXXXXXXXX:/app# python Python 3.7.7 (default, Mar 23 2020, 22:36:06) [GCC 7.3.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.

import torch print(torch.cuda.is_available()) False

mellahysf commented 3 years ago

It resolved by installing nvidia-container-toolkit to run GPU from docker.