diff --git a/bindings/torch/setup.py b/bindings/torch/setup.py
index 9385914a..60cf64cc 100644
--- a/bindings/torch/setup.py
+++ b/bindings/torch/setup.py
@@ -2,7 +2,7 @@
 
 import re
 from setuptools import setup
-from pkg_resources import parse_version
+from setuptools.extern.packaging.version import parse as parse_version
 import subprocess
 import shutil
 import sys