About Faiss

[lixinying0727]

Traceback (most recent call last):
File "bin/run_patchcore.py", line 435, in
main()
File "/home/xpu107/下载/yes/envs/PatchCore/lib/python3.8/site-packages/click/core.py", line 1161, in call
return self.main(*args, **kwargs)
File "/home/xpu107/下载/yes/envs/PatchCore/lib/python3.8/site-packages/click/core.py", line 1082, in main
rv = self.invoke(ctx)
File "/home/xpu107/下载/yes/envs/PatchCore/lib/python3.8/site-packages/click/core.py", line 1729, in invoke
return _process_result(rv)
File "/home/xpu107/下载/yes/envs/PatchCore/lib/python3.8/site-packages/click/core.py", line 1666, in _process_result
value = ctx.invoke(self._result_callback, value, **ctx.params)
File "/home/xpu107/下载/yes/envs/PatchCore/lib/python3.8/site-packages/click/core.py", line 788, in invoke
return __callback(*args, **kwargs)
File "bin/run_patchcore.py", line 97, in run
PatchCore.fit(dataloaders["training"])
File "/media/xpu107/LJG/patchcore-inspection-main/src/patchcore/patchcore.py", line 154, in fit
self._fill_memory_bank(training_data)
File "/media/xpu107/LJG/patchcore-inspection-main/src/patchcore/patchcore.py", line 177, in _fill_memory_bank
self.anomaly_scorer.fit(detection_features=[features])
File "/media/xpu107/LJG/patchcore-inspection-main/src/patchcore/common.py", line 330, in fit
self.nn_method.fit(self.detection_features)
File "/media/xpu107/LJG/patchcore-inspection-main/src/patchcore/common.py", line 59, in fit
self.search_index = self._create_index(features.shape[-1])
File "/media/xpu107/LJG/patchcore-inspection-main/src/patchcore/common.py", line 45, in _create_index
return faiss.GpuIndexFlatL2(
AttributeError: module 'faiss' has no attribute 'GpuIndexFlatL2'

[xiaoqingling]

可能是你先装了faiss-cpu，然后再装的faiss-gpu，然后使用gpu推理时优先读到了cpu的包，建议重装环境，然后只装gpu的包，我是这么解决的