安装 faiss

# Make sure you have CUDA installed before installing faiss-gpu, # otherwise it falls back to CPU version conda install faiss-gpu -c pytorch # [DEFAULT]For CUDA8.0, comes with cudatoolkit8.0 conda install faiss-gpu cuda90 -c pytorch # For CUDA9.0 conda install faiss-gpu cuda91 -c pytorch # For CUDA9.1 # cuda90/cuda91 shown above is a feature, it doesn t install CUDA for you.

参考链接

https://github.com/fchollet/deep-learning-models/releases/download/v0.1/vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5