kadirnar commited on
Commit
29484db
·
verified ·
1 Parent(s): df018bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ def install_apex():
15
  cmd = [
16
  'pip', 'install', '-v', '--disable-pip-version-check', '--no-cache-dir',
17
  '--no-build-isolation', '--config-settings', '--build-option=--cpp_ext',
 
18
  '--config-settings', '--build-option=--cuda_ext', 'git+https://github.com/NVIDIA/apex.git'
19
  ]
20
  try:
 
15
  cmd = [
16
  'pip', 'install', '-v', '--disable-pip-version-check', '--no-cache-dir',
17
  '--no-build-isolation', '--config-settings', '--build-option=--cpp_ext',
18
+ '--global-option="--cpp_ext"','--global-option="--cuda_ext"',
19
  '--config-settings', '--build-option=--cuda_ext', 'git+https://github.com/NVIDIA/apex.git'
20
  ]
21
  try: