From ff6ae5a622b05b5f45915168ef8135f0dc9ed713 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Fri, 6 Mar 2020 14:42:40 -0600
Subject: [PATCH] ...

---
 pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pipeline.py b/pipeline.py
index e6283fd..dfa0331 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -150,7 +150,7 @@ class Components :
 		else:
 			_args['gpu']  = 0
 		_args['num_gpu'] 	= 1
-		
+		os.environ['CUDA_VISIBLE_DEVICES'] = str(args['gpu']) 
 		_args['no_value']= args['no_value']
 		# MAX_ROWS = args['max_rows']  	if 'max_rows' in args else 0
 		PART_SIZE = int(args['part_size']) if 'part_size' in args else 8