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天

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值