pytorch的各种错误集锦
程序员文章站
2022-05-30 16:29:57
...
安装类
faiss
错误提示:No module named '_swigfaiss'
原因:
没有安装依赖包,而且我直接用pip install faiss
安装,没采用官方推荐的conda
方法
解决方法:
1.删除原本安装的faiss
2.conda install mkl, openblas swig
3.用conda安装faiss
# CPU版本
conda install faiss-cpu -c pytorch
# GPU版本
conda install faiss-gpu cudatoolkit=8.0 -c pytorch # For CUDA8
conda install faiss-gpu cudatoolkit=9.0 -c pytorch # For CUDA9
conda install faiss-gpu cudatoolkit=10.0 -c pytorch # For CUDA10
备注
faiss只能安装在Linux和Mac系统,非conda安装请参照官网