Skip to content

Commit

Permalink
Issue yell#8 Add setup.py CPU
Browse files Browse the repository at this point in the history
  • Loading branch information
SrMouraSilva committed Mar 6, 2019
1 parent 6a10b61 commit 4932b80
Show file tree
Hide file tree
Showing 3 changed files with 49 additions and 7 deletions.
32 changes: 27 additions & 5 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,14 +1,36 @@
# Python build
*.pyc

# Tensorflow files
*.json

# Pycharm Project
.idea/*
.idea

# Virtualenv
.venv/*
boltzmann_machines/test_rbm_1/*
boltzmann_machines/test_rbm_2/*

models/*
!models/fetch_models.sh
data/*.npy
data/mnist/
data/cifar-10-batches-py/

# Notebooks
notebooks/.ipynb_checkpoints/*
models/*
!models/fetch_models.sh

# LaTex files
tex/*.swp
.idea

# python setup.py develop
# python setup_cpu.py develop
boltzmann_machines.egg-info/

# Coverage
.coverage

# Research projects
.tests/
boltzmann_machines/test_rbm_1/*
boltzmann_machines/test_rbm_2/*
5 changes: 3 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
setup(name='boltzmann_machines',
version='0.1',
url='https://github.com/monsta-hd/boltzmann-machines',
install_requires=[ 'tensorflow-gpu>=1.3.0',
install_requires=['tensorflow-gpu>=1.3.0',
'scipy>=0.17',
'keras>=2.0.8',
'matplotlib>=1.5.3',
Expand All @@ -13,6 +13,7 @@
'numpy>=1.13.0',
'scikit-learn>=0.19.0',
'seaborn>=0.8.1',
'tqdm>=4.14.0'
'tqdm>=4.14.0',
'coverage>=4.5.0'
],
packages=find_packages())
19 changes: 19 additions & 0 deletions setup_cpu.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
from setuptools import setup
from setuptools import find_packages

setup(name='boltzmann_machines',
version='0.1',
url='https://github.com/monsta-hd/boltzmann-machines',
install_requires=[ 'tensorflow>=1.13.0',
'scipy>=0.17',
'keras>=2.0.8',
'matplotlib>=1.5.3',
'nose-exclude>=0.5.0',
'nose>=1.3.4',
'numpy>=1.13.0',
'scikit-learn>=0.19.0',
'seaborn>=0.8.1',
'tqdm>=4.14.0',
'coverage>=4.5.0'
],
packages=find_packages())

0 comments on commit 4932b80

Please sign in to comment.