Use nvidia-ml-py3 for python3 compatibility

1 files changed, 1 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 852397b7..f44358c5 100755
--- a/setup.py
+++ b/setup.py
@@ -88,7 +88,7 @@ setup(
                    'influxdb>=1.0.0', 'kafka-python', 'pika', 'potsdb',
                    'prometheus_client', 'pyzmq', 'statsd'],
         'folders': ['scandir'],  # python_version<"3.5"
-        'gpu': ['nvidia-ml-py'],  # python_version=="2.7"
+        'gpu': ['nvidia-ml-py3'],  # python_version=="2.7"
         'graph': ['pygal'],
         'ip': ['netifaces'],
         'raid': ['pymdstat'],