TorchPQ
Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
파일 탐색기
최종 버전 다운로드 (.zip)- settings.json
- ivf[16384]_pq[16]_recall@1.png
- ivf[16384]_pq[16]_recall@10.png
- ivf[16384]_pq[16]_recall@100.png
- ivf[16384]_pq[32]_recall@1.png
- ivf[16384]_pq[32]_recall@10.png
- ivf[16384]_pq[32]_recall@100.png
- ivf[16384]_pq[64]_recall@1.png
- ivf[16384]_pq[64]_recall@10.png
- ivf[16384]_pq[64]_recall@100.png
- ivf[16384]_pq[8]_recall@1.png
- ivf[16384]_pq[8]_recall@10.png
- ivf[16384]_pq[8]_recall@100.png
- ivf[4096]_pq[16]_recall@1.png
- ivf[4096]_pq[16]_recall@10.png
- ivf[4096]_pq[16]_recall@100.png
- ivf[4096]_pq[32]_recall@1.png
- ivf[4096]_pq[32]_recall@10.png
- ivf[4096]_pq[32]_recall@100.png
- ivf[4096]_pq[64]_recall@1.png
- ivf[4096]_pq[64]_recall@10.png
- ivf[4096]_pq[64]_recall@100.png
- ivf[4096]_pq[8]_recall@1.png
- ivf[4096]_pq[8]_recall@10.png
- ivf[4096]_pq[8]_recall@100.png
- faiss_ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_residual_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- README.md
- T4_7_12_2021.zip
- n_clusters=16384 n_features=256.png
- n_clusters=256 n_features=256.png
- n_data=1000000 n_clusters=1024.png
- n_data=1000000 n_features=128.png
- README.md
- ivf[16384]_pq[16]_recall@1.png
- ivf[16384]_pq[16]_recall@10.png
- ivf[16384]_pq[16]_recall@100.png
- ivf[16384]_pq[32]_recall@1.png
- ivf[16384]_pq[32]_recall@10.png
- ivf[16384]_pq[32]_recall@100.png
- ivf[16384]_pq[64]_recall@1.png
- ivf[16384]_pq[64]_recall@10.png
- ivf[16384]_pq[64]_recall@100.png
- ivf[16384]_pq[8]_recall@1.png
- ivf[16384]_pq[8]_recall@10.png
- ivf[16384]_pq[8]_recall@100.png
- ivf[4096]_pq[16]_recall@1.png
- ivf[4096]_pq[16]_recall@10.png
- ivf[4096]_pq[16]_recall@100.png
- ivf[4096]_pq[32]_recall@1.png
- ivf[4096]_pq[32]_recall@10.png
- ivf[4096]_pq[32]_recall@100.png
- ivf[4096]_pq[64]_recall@1.png
- ivf[4096]_pq[64]_recall@10.png
- ivf[4096]_pq[64]_recall@100.png
- ivf[4096]_pq[8]_recall@1.png
- ivf[4096]_pq[8]_recall@10.png
- ivf[4096]_pq[8]_recall@100.png
- faiss_ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_residual_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- ivf[16384]_pq[16]_recall@1.png
- ivf[16384]_pq[16]_recall@10.png
- ivf[16384]_pq[16]_recall@100.png
- ivf[16384]_pq[32]_recall@1.png
- ivf[16384]_pq[32]_recall@10.png
- ivf[16384]_pq[32]_recall@100.png
- ivf[16384]_pq[64]_recall@1.png
- ivf[16384]_pq[64]_recall@10.png
- ivf[16384]_pq[64]_recall@100.png
- ivf[16384]_pq[8]_recall@1.png
- ivf[16384]_pq[8]_recall@10.png
- ivf[16384]_pq[8]_recall@100.png
- ivf[4096]_pq[16]_recall@1.png
- ivf[4096]_pq[16]_recall@10.png
- ivf[4096]_pq[16]_recall@100.png
- ivf[4096]_pq[32]_recall@1.png
- ivf[4096]_pq[32]_recall@10.png
- ivf[4096]_pq[32]_recall@100.png
- ivf[4096]_pq[64]_recall@1.png
- ivf[4096]_pq[64]_recall@10.png
- ivf[4096]_pq[64]_recall@100.png
- ivf[4096]_pq[8]_recall@1.png
- ivf[4096]_pq[8]_recall@10.png
- ivf[4096]_pq[8]_recall@100.png
- faiss_ivf[16]_pq[16384]_sift1m.json
- faiss_ivf[16]_pq[4096]_sift1m.json
- faiss_ivf[32]_pq[16384]_sift1m.json
- faiss_ivf[32]_pq[4096]_sift1m.json
- faiss_ivf[64]_pq[16384]_sift1m.json
- faiss_ivf[64]_pq[4096]_sift1m.json
- faiss_ivf[8]_pq[16384]_sift1m.json
- faiss_ivf[8]_pq[4096]_sift1m.json
- ivf[16]_pq[16384]_residual_sift1m.json
- ivf[16]_pq[16384]_sift1m.json
- ivf[16]_pq[4096]_cublas.json
- ivf[16]_pq[4096]_residual_sift1m.json
- ivf[16]_pq[4096]_sift1m.json
- ivf[16]_pq[4096]_smart.json
- ivf[32]_pq[16384]_residual_sift1m.json
- ivf[32]_pq[16384]_sift1m.json
- ivf[32]_pq[4096]_residual_sift1m.json
- ivf[32]_pq[4096]_sift1m.json
- ivf[64]_pq[16384]_residual_sift1m.json
- ivf[64]_pq[16384]_sift1m.json
- ivf[64]_pq[4096]_residual_sift1m.json
- ivf[64]_pq[4096]_sift1m.json
- ivf[8]_pq[16384]_residual_sift1m.json
- ivf[8]_pq[16384]_sift1m.json
- ivf[8]_pq[4096]_cublas.json
- ivf[8]_pq[4096]_residual_sift1m.json
- ivf[8]_pq[4096]_sift1m.json
- ivf[8]_pq[4096]_smart.json
- T4.zip
- ivf[16384]_pq[16]_recall@1.png
- ivf[16384]_pq[16]_recall@10.png
- ivf[16384]_pq[16]_recall@100.png
- ivf[16384]_pq[32]_recall@1.png
- ivf[16384]_pq[32]_recall@10.png
- ivf[16384]_pq[32]_recall@100.png
- ivf[16384]_pq[64]_recall@1.png
- ivf[16384]_pq[64]_recall@10.png
- ivf[16384]_pq[64]_recall@100.png
- ivf[16384]_pq[8]_recall@1.png
- ivf[16384]_pq[8]_recall@10.png
- ivf[16384]_pq[8]_recall@100.png
- ivf[4096]_pq[16]_recall@1.png
- ivf[4096]_pq[16]_recall@10.png
- ivf[4096]_pq[16]_recall@100.png
- ivf[4096]_pq[32]_recall@1.png
- ivf[4096]_pq[32]_recall@10.png
- ivf[4096]_pq[32]_recall@100.png
- ivf[4096]_pq[64]_recall@1.png
- ivf[4096]_pq[64]_recall@10.png
- ivf[4096]_pq[64]_recall@100.png
- ivf[4096]_pq[8]_recall@1.png
- ivf[4096]_pq[8]_recall@10.png
- ivf[4096]_pq[8]_recall@100.png
- faiss_ivf[16]_pq[16384]_sift1m.json
- faiss_ivf[16]_pq[4096]_sift1m.json
- faiss_ivf[32]_pq[16384]_sift1m.json
- faiss_ivf[32]_pq[4096]_sift1m.json
- faiss_ivf[64]_pq[16384]_sift1m.json
- faiss_ivf[64]_pq[4096]_sift1m.json
- faiss_ivf[8]_pq[16384]_sift1m.json
- faiss_ivf[8]_pq[4096]_sift1m.json
- ivf[16]_pq[16384]_residual_sift1m.json
- ivf[16]_pq[16384]_sift1m.json
- ivf[16]_pq[16384]_smart_sift1m.json
- ivf[16]_pq[4096]_cublas.json
- ivf[16]_pq[4096]_residual_sift1m.json
- ivf[16]_pq[4096]_sift1m.json
- ivf[16]_pq[4096]_smart.json
- ivf[16]_pq[4096]_smart_sift1m.json
- ivf[32]_pq[16384]_residual_sift1m.json
- ivf[32]_pq[16384]_sift1m.json
- ivf[32]_pq[16384]_smart_sift1m.json
- ivf[32]_pq[4096]_residual_sift1m.json
- ivf[32]_pq[4096]_sift1m.json
- ivf[32]_pq[4096]_smart_sift1m.json
- ivf[64]_pq[16384]_residual_sift1m.json
- ivf[64]_pq[16384]_sift1m.json
- ivf[64]_pq[16384]_smart_sift1m.json
- ivf[64]_pq[4096]_residual_sift1m.json
- ivf[64]_pq[4096]_sift1m.json
- ivf[64]_pq[4096]_smart_sift1m.json
- ivf[8]_pq[16384]_residual_sift1m.json
- ivf[8]_pq[16384]_sift1m.json
- ivf[8]_pq[16384]_smart_sift1m.json
- ivf[8]_pq[4096]_cublas.json
- ivf[8]_pq[4096]_residual_sift1m.json
- ivf[8]_pq[4096]_sift1m.json
- ivf[8]_pq[4096]_smart.json
- ivf[8]_pq[4096]_smart_sift1m.json
- T4.zip
- ivf[16384]_pq[16]_recall@1.png
- ivf[16384]_pq[16]_recall@10.png
- ivf[16384]_pq[16]_recall@100.png
- ivf[16384]_pq[32]_recall@1.png
- ivf[16384]_pq[32]_recall@10.png
- ivf[16384]_pq[32]_recall@100.png
- ivf[16384]_pq[64]_recall@1.png
- ivf[16384]_pq[64]_recall@10.png
- ivf[16384]_pq[64]_recall@100.png
- ivf[16384]_pq[8]_recall@1.png
- ivf[16384]_pq[8]_recall@10.png
- ivf[16384]_pq[8]_recall@100.png
- ivf[4096]_pq[16]_recall@1.png
- ivf[4096]_pq[16]_recall@10.png
- ivf[4096]_pq[16]_recall@100.png
- ivf[4096]_pq[32]_recall@1.png
- ivf[4096]_pq[32]_recall@10.png
- ivf[4096]_pq[32]_recall@100.png
- ivf[4096]_pq[64]_recall@1.png
- ivf[4096]_pq[64]_recall@10.png
- ivf[4096]_pq[64]_recall@100.png
- ivf[4096]_pq[8]_recall@1.png
- ivf[4096]_pq[8]_recall@10.png
- ivf[4096]_pq[8]_recall@100.png
- faiss_ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_residual_sift1m.json
- ivf[8, 16, 32, 64]_pq[4096, 16384]_sift1m.json
- T4_7_11_2021.zip
- README.md
- T4_7_12_2021.zip
- 1.png
- 2.png
- 3.png
- 1.png
- 2.png
- 3.png
- 4.png
- 6.png
- 7.png
- 8.png
- CellContainerConfig.yml
- FlatContainerConfig.yml
- FlatContainerGroupConfig.yml
- CellContainerTestCase.py
- CustomTestCase.py
- FlatContainerGroupTestCase.py
- FlatContainerTestCase.py
- __init__.py
- KMeans.py
- KMeansOld.py
- MinibatchKMeans.py
- MultiKMeans.py
- MultiKMeansOld.py
- __init__.py
- BaseCodec.py
- PQCodec.py
- SQCodec.py
- VQCodec.py
- __init__.py
- BaseContainer.py
- CellContainer.py
- DistributedCellContainer.py
- FlatContainer.py
- FlatContainerGroup.py
- __init__.py
- IVFPQTopk.py
- Topk.py
- __init__.py
- FlatIndex.py
- IVFPQIndex.py
- IVFPQRIndex.py
- bmm.cu
- bmm_helpers.cu
- compute_centroids.cu
- compute_product.cu
- distirbuted_ivfpq_top1.cu
- distributed_ivfpq_topk.cu
- get_address_by_id.cu
- get_div_by_address.cu
- get_div_by_address_v2.cu
- get_ioa.cu
- get_write_address.cu
- get_write_address_v2.cu
- ivfpq4_top1_v1.cu
- ivfpq4_top1_v1_failed.cu
- ivfpq_top1.cu
- ivfpq_top1_v2.cu
- ivfpq_top1_v2_failed.cu
- ivfpq_topk.cu
- ivfpq_topk_v1.cu
- ivfpq_topk_v2.cu
- max_sim.cu
- mbmm.cu
- minbmm.cu
- pq_decode.cu
- top1_select.cu
- top32_select.cu
- topk_select.cu
- topk_select_v2.cu
- topk_select_v2_maybe_later.cu
- topkbmm.cu
- __init__.py
- BMMCuda.py
- ComputeCentroidsCuda.py
- ComputeProductCuda.py
- CustomKernel.py
- default_device.py
- DistributedIVFPQTop1Cuda.py
- DistributedIVFPQTopkCuda.py
- GetAddressByIDCuda.py
- GetDivByAddressCuda.py
- GetDivByAddressV2Cuda.py
- GetIOACuda.py
- GetWriteAddressCuda.py
- GetWriteAddressV2Cuda.py
- IVFPQ4Top1Cuda_v1.py
- IVFPQTop1Cuda.py
- IVFPQTop1Cuda_v2.py
- IVFPQTopkCuda.py
- MaxSimCuda.py
- MBMMCuda.py
- MinBMMCuda.py
- PQDecodeCuda.py
- Top1SelectCuda.py
- Top32SelectCuda.py
- TopkBMMCuda.py
- TopkSelectCuda.py
- TopkSelectCuda_v2.py
- __init__.py
- IVFPQ.py
- IVFPQBase.py
- IVFPQR.py
- IVFPQTopk.py
- PQ.py
- SQ.py
- __init__.py
- OPQ.py
- PCA.py
- __init__.py
- _version.py
- CustomModule.py
- metric.py
- util.py
- .gitignore
- CITATION.cff
- CODE_OF_CONDUCT.md
- LICENSE
- MANIFEST.in
- profile.sh
- README.md
- setup.py
# 설치 가이드
1. 코드 내려받기
git clone https://github.com/DeMoriarty/TorchPQ
깃허브에서 프로젝트 코드 전체를 내 컴퓨터로 내려받습니다.
cd TorchPQ
방금 내려받은 프로젝트 폴더 안으로 이동합니다.
2. 공식 설치 스크립트
쉬움 추천사전 준비물
- Python 3 pip 명령어를 쓰려면 Python이 필요합니다.
pip install cupy-cuda110
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install cupy-cuda111
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install cupy-cuda112
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install torchpq
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
설치 후 새 터미널을 열고, 프로그램의 버전 확인 명령(예: --version)으로 정상 설치됐는지 확인하세요.
이 레포의 README에 적힌 실제 명령어를 그대로 가져왔습니다.
3. Python
쉬움사전 준비물
pip install cupy-cuda110
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install cupy-cuda111
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install cupy-cuda112
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install torchpq
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
에러 메시지 없이 실행되고 터미널에 안내 문구가 출력되면 정상입니다.
이 레포의 README에 적힌 실제 명령어를 그대로 가져왔습니다.
// repository documentation
Was this content helpful?
(0 ratings)
