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.
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.