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才行,..
原创
2022-03-28 18:48:02 ·
2691 阅读 ·
1 评论