Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/master' into einsum
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed Jul 15, 2022
2 parents 5ab5c0d + c7451a9 commit e168125
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 3 deletions.
5 changes: 4 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -110,4 +110,7 @@ venv.bak/
design-einops.txt

# logo and video generation
logo
logo

# code of internal checks
_uncommited_explorations
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
classifiers=[
'Intended Audience :: Science/Research',
'Programming Language :: Python :: 3 ',
'License :: OSI Approved :: MIT License',
],
keywords='deep learning, neural networks, tensor manipulation, machine learning, '
'scientific computations, einops',
Expand Down
8 changes: 6 additions & 2 deletions test.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,12 @@ def run(cmd, **env):


# check we have nvidia-smi
output, _ = Popen('which nvidia-smi'.split(' '), stdout=PIPE).communicate()
have_cuda = b'nvidia' in output
import shutil
have_cuda = False
if shutil.which('nvidia-smi') is not None:
output, _ = Popen('nvidia-smi'.split(' '), stdout=PIPE).communicate()
if b'failed because' not in output:
have_cuda = True

# install cupy. It can't be installed without cuda available (with compilers).
skip_cupy = not have_cuda
Expand Down

0 comments on commit e168125

Please sign in to comment.