search

Tencent / NeuralNLP-NeuralClassifier

An Open-source Neural Hierarchical Multi-label Text Classification Toolkit
Other
1.85k stars 406 forks source link

RuntimeError: CUDA error: invalid device ordinal #21

Closed SlinZhang closed 5 years ago

SlinZhang commented 5 years ago

"device": "cuda:1" in train.json get RuntimeError: CUDA error: invalid device ordinal