Skip to content

Instantly share code, notes, and snippets.

@mdouze

mdouze/gist:df42bd8ad8c74c42fedd3fd1043e83b7 Secret

Created April 4, 2022 07:45
Show Gist options
  • Save mdouze/df42bd8ad8c74c42fedd3fd1043e83b7 to your computer and use it in GitHub Desktop.
Save mdouze/df42bd8ad8c74c42fedd3fd1043e83b7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
(faiss_1.7.2) matthijs@devfair0144:~/faiss_versions/issue_2273/FAISS_GPU_TEST$ python main.py
tensor([[7.6294e-06, 4.8415e+00, 4.9375e+00, 6.1092e+00, 6.7700e+00],
[3.8147e-06, 4.4300e+00, 4.7558e+00, 8.6782e+00, 8.7311e+00],
[0.0000e+00, 4.4747e+00, 4.5966e+00, 5.5469e+00, 7.5937e+00],
[0.0000e+00, 3.5889e+00, 5.0067e+00, 5.1544e+00, 5.3525e+00],
[7.6294e-06, 4.4523e+00, 4.4754e+00, 5.3311e+00, 5.6527e+00]])
tensor([[7.6294e-06, 4.8415e+00, 4.9375e+00, 6.1092e+00, 6.7700e+00],
[7.6294e-06, 4.4300e+00, 4.7558e+00, 8.6782e+00, 8.7311e+00],
[0.0000e+00, 4.4747e+00, 4.5966e+00, 5.5469e+00, 7.5937e+00],
[0.0000e+00, 3.5889e+00, 5.0067e+00, 5.1544e+00, 5.3525e+00],
[3.8147e-06, 4.4523e+00, 4.4754e+00, 5.3312e+00, 5.6527e+00]],
device='cuda:0')
tensor([[7.6294e-06, 4.8415e+00, 4.9375e+00, 6.1092e+00, 6.7700e+00],
[7.6294e-06, 4.4300e+00, 4.7558e+00, 8.6782e+00, 8.7311e+00],
[0.0000e+00, 4.4747e+00, 4.5966e+00, 5.5469e+00, 7.5937e+00],
[0.0000e+00, 3.5889e+00, 5.0067e+00, 5.1544e+00, 5.3525e+00],
[3.8147e-06, 4.4523e+00, 4.4754e+00, 5.3312e+00, 5.6527e+00]],
device='cuda:0')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment