diff --git a/setup.py b/setup.py
index 0cb39c8..a034a38 100644
--- a/setup.py
+++ b/setup.py
@@ -68,7 +68,7 @@ def findcuda():
 cuda_headers = find(curdir, lambda file: file.endswith(".cuh"), True)
 headers = find(curdir, lambda file: file.endswith(".h"), True)
 
-libaten = find(torch_dir, re.compile("libaten", re.IGNORECASE).search, False)
+libaten = list(set(find(torch_dir, re.compile("libaten.*(so)\Z", re.IGNORECASE).match, True)))
 aten_h = find(torch_dir, re.compile("aten.h", re.IGNORECASE).search, False)
 
 include_dirs = [os.path.dirname(os.path.dirname(aten_h))]
@@ -81,7 +81,7 @@ def findcuda():
 assert libaten, "Could not find PyTorch's libATen."
 assert aten_h, "Could not find PyTorch's ATen header."
 
-library_dirs.append(os.path.dirname(libaten))
+library_dirs.append(os.path.dirname(libaten[0]))
 
 #create some places to collect important things
 object_files = []