GeForce RTX 3090 GPU with PyTorch 算力不匹配问题(也是spacy3.2的prefer_gpu()总是失败的根本问题)

我的环境是anaconda3下的python3.6.5的虚拟环境,且已经安装了cuda11.3,安装了spacy[cu113]版本,也默认安装了依赖项pytorch的1.10也对应cu113版本,但是spacy运用gpu训练模型总是出错,最终发现是spacy依赖的pytorch的算力支持最高到sm70,不能支持GeForce RTX 3090 的8.6的算力。花了3天的时间,卸载安装了无数次,都要崩溃了。最终发现 pytorch的依赖项 torchvision和torchaudio都要更新到cu113才行,参考https://blog.csdn.net/a563562675/article/details/121656894

 pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html  更新了相关库,终于可以用了,折腾了3天

你可能感兴趣的:(spacy,3090显卡,python)