colin1842 commited on
Commit
46c2566
·
1 Parent(s): 8fad1c3
Files changed (2) hide show
  1. .gitignore +2 -0
  2. script.py +3 -1
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ packages
2
+ __pycache__
script.py CHANGED
@@ -148,7 +148,8 @@ def setup_environment():
148
  hoho.setup()
149
 
150
  # Download required packages
151
- packages_to_download = ['torch==2.0.1', 'torchvision==0.15.2', 'torchaudio==2.0.2']
 
152
  download_packages(packages_to_download, folder='packages/torch')
153
 
154
  # Install packages from local files
@@ -161,6 +162,7 @@ def setup_environment():
161
  subprocess.check_call([sys.executable, "-m", "pip", "install", "scipy"])
162
  subprocess.check_call([sys.executable, "-m", "pip", "install", "open3d"])
163
  subprocess.check_call([sys.executable, "-m", "pip", "install", "easydict"])
 
164
 
165
  # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
166
  # if os.path.isdir(pc_util_path):
 
148
  hoho.setup()
149
 
150
  # Download required packages
151
+ # pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
152
+ packages_to_download = ['torch==1.13.1', 'torchvision==0.14.1', 'torchaudio==0.13.1']
153
  download_packages(packages_to_download, folder='packages/torch')
154
 
155
  # Install packages from local files
 
162
  subprocess.check_call([sys.executable, "-m", "pip", "install", "scipy"])
163
  subprocess.check_call([sys.executable, "-m", "pip", "install", "open3d"])
164
  subprocess.check_call([sys.executable, "-m", "pip", "install", "easydict"])
165
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python"])
166
 
167
  # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
168
  # if os.path.isdir(pc_util_path):