Spacy not working for GPU on cuda 117 #11272
-
How to reproduce the behaviourErrors out on line 98 of the attached code (it is a .py but github won't let me attach unless it's a .txt) with traceback: Your Environment
My nvcc verison is: My cuda verison is: 11.7 My nvidia driver version is: 516.59 My CUDNN version is: v 8.4.1.50 Running on NVIDIA rtx 2060 it is the only card attached The only reason I have torch for cuda 11.6 is because there is no pip version for cuda 11.7 available and I was unable to build it from source |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 6 replies
-
This looks more like it might be a problem with CUDA or torch rather than spacy itself. The attached file of pip packages was empty, can you provide the output of You can also test a simpler example to see if it works or provides a more informative error message: import spacy
spacy.require_gpu()
nlp = spacy.load("en_core_web_trf")
doc = nlp("This is a test sentence.") |
Beta Was this translation helpful? Give feedback.
This looks more like it might be a problem with CUDA or torch rather than spacy itself. The attached file of pip packages was empty, can you provide the output of
pip freeze
?You can also test a simpler example to see if it works or provides a more informative error message: