Made python2 compatible

Author: fbcotter

py3nvml/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
 from py3nvml import py3nvml
 from py3nvml import nvidia_smi
 from py3nvml.utils import grab_gpus

py3nvml/py3nvml.py

@@ -381,7 +381,10 @@ class struct_c_nvmlUnit_t(Structure):
 # Convert bytes objects to strings or leave untouched
 def bytes_to_str(s):
     if type(s) is bytes:
-        return str(s, 'utf-8')
+        try:
+            return str(s, 'utf-8')
+        except TypeError:  # Can get this if running python 2
+            return str(s)
     else:
         return s
 

py3nvml/utils.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import
 from __future__ import print_function
+from __future__ import division
 
 import logging
 import os

scripts/py3smi

@@ -1,4 +1,7 @@
 #! /usr/bin/env python
+from __future__ import print_function
+from __future__ import division
+from __future__ import absolute_import
 
 from py3nvml.py3nvml import *
 from datetime import datetime