andy-wyx commited on
Commit
ad74432
·
1 Parent(s): 40577ea

update pre-installed dependencies

Browse files
Files changed (2) hide show
  1. env.py +1 -9
  2. requirements.txt +1 -0
env.py CHANGED
@@ -22,9 +22,6 @@ def config_env():
22
 
23
  name_to_command = {'segment_anything':'git+https://github.com/facebookresearch/segment-anything.git',
24
  'panopticapi':'git+https://github.com/cocodataset/panopticapi.git',
25
- #TODO
26
- 'torch':'torch --index-url https://download.pytorch.org/whl/cu118',
27
- 'torchvision':'torchvision --index-url https://download.pytorch.org/whl/cu118',
28
  }
29
 
30
  env_name = os.path.basename(sys.prefix)
@@ -33,12 +30,7 @@ def config_env():
33
  package_spec = f"{package}=={version}" if version else package
34
  package_spec = name_to_command[package_spec] if package_spec in name_to_command else package_spec
35
  if not is_pkg_installed(package):
36
- #TODO
37
- if package=='torch' or 'torchvision' or 'tensorflow':
38
- # TMPDIR = 'home/username/tmp' for temporary compilation to prevent disk partition storage limit on zmachine
39
- subprocess.call(f"TMPDIR='../../../tmp/' pip install {package_spec}",shell=True)
40
- else:
41
- subprocess.call(f"pip install {package_spec}".split())
42
  else:
43
  print(f"{package_spec} is already installed.")
44
  else:
 
22
 
23
  name_to_command = {'segment_anything':'git+https://github.com/facebookresearch/segment-anything.git',
24
  'panopticapi':'git+https://github.com/cocodataset/panopticapi.git',
 
 
 
25
  }
26
 
27
  env_name = os.path.basename(sys.prefix)
 
30
  package_spec = f"{package}=={version}" if version else package
31
  package_spec = name_to_command[package_spec] if package_spec in name_to_command else package_spec
32
  if not is_pkg_installed(package):
33
+ subprocess.call(f"pip install {package_spec}".split())
 
 
 
 
 
34
  else:
35
  print(f"{package_spec} is already installed.")
36
  else:
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  subprocess
2
  importlib.metadata
3
  gradio
 
1
+ python==3.9.12
2
  subprocess
3
  importlib.metadata
4
  gradio