Merge pull request #8 from dingguanglei/development

Development
dingguanglei · Oct 25, 2018 · 439a60d · 439a60d
2 parents e435196 + d11c477
commit 439a60d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,10 @@
+nvidia_ml_py3==7.352.0
+torchvision==0.2.1
+psutil==5.4.6
+tqdm==4.23.4
+scipy==1.1.0
+torch==0.4.1
+pandas==0.23.1
+numpy==1.14.5
+imageio==2.4.1
+tensorboardX==1.4
diff --git a/setup.py b/setup.py
@@ -21,8 +21,12 @@
         'tqdm>=4.23.4',
         'pandas>=0.23.0',
         "tensorboardX>=1.4",
-        'tensorboard>=1.7.0'
-    ],
+        'tensorboard>=1.7.0',
+        "nvidia_ml_py3>=7.352.0",
+        "scipy>=1.1.0",
+        "pandas>=0.23.1",
+        'numpy>=1.14.5',
+        'imageio>=2.4.1'],
 
     # # 添加这个选项，在windows下Python目录的scripts下生成exe文件
     # # 注意：模块与函数之间是冒号: