alpha 3

Author: narendasan

nvdocker/nvdocker.py

@@ -69,11 +69,23 @@ def run(self, image, cmd="", **kwargs):
         #defaults
         config = {}
         environment = {}
-        for arg in kwargs:
+       for arg in kwargs:
             if arg == "driver_capabilities":
                 environment["NVIDIA_DRIVER_CAPABILITIES"] = kwargs["driver_capabilities"]
             elif arg == "visible_devices" in kwargs:
-                environment["NVIDIA_VISIBLE_DEVICES"] = kwargs["visible_devices"]
+                vis_devices = ""
+                if type(kwargs["visible_devices"]) is list:
+                    if len(kwargs["visible_devices"]) == 1:
+                        vis_devices = str(kwargs["visible_devices"][0])
+                    else:
+                        for dev in kwargs["visible_devices"]:
+                            vis_devices += dev + ','
+                        vis_devices = vis_devices[:-1]
+                elif type(kwargs["visible_devices"]) is str:
+                    vis_devices = kwargs["visible_device"]
+                elif type(kwargs["visible_devices"]) is int:
+                    vis_devices = str(kwargs["visible_devices"])
+                environment["NVIDIA_VISIBLE_DEVICES"] = vis_devices
             elif arg == "disable_require" in kwargs:
                 environment["NVIDIA_DISABLE_REQUIRE"] = kwargs["disable_require"]
             elif arg == "require":

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = nvdocker
-version = 0.0.2a2
+version = 0.0.2a3
 author = ACM@UIUC
 author-email = acm@illinois.edu
 summary = Python interface for NVIDIA Docker 

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name='nvdocker',
-    version='0.0.2a2',  
+    version='0.0.2a3',  
     description='nvdocker is library built on top of the docker-py python sdk to build and run docker containers using nvidia-docker. Targets nvidia-docker2', 
     long_description=long_description, 
     url='https://github.com/acm-uiuc/nvdocker',