一、问题背景
启用pytorch的cuda支持时,发现程序运行不了,该原因是由于,pytorch默认支持的是cpu,安装支持gpu的版本即可。
报错如下:
Torch not compiled with CUDA enabled
二、解决方案
- 卸载pytorch
pip uninstall torch torchvision torchaudio
2.安装支持gpu的pytorch(2023年9月15日时,pytorch版本是2.0.1,官方推荐下载的cuda117和cuda118,具体的安装命令,需要结合pytorch官网的实时文档)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
三、搞定
再次调试text2vec,配置cuda模式,已经能正常运转了。
【备注】:text2vec 开启cuda的方法如下:
from text2vec import SentenceModel text2vec.similarity.device='cuda'