colin1842 commited on
Commit
c482d53
·
1 Parent(s): fffb15c

update packages

Browse files
Files changed (2) hide show
  1. .gitignore +0 -1
  2. script.py +9 -6
.gitignore CHANGED
@@ -1,2 +1 @@
1
- packages
2
  __pycache__
 
 
1
  __pycache__
script.py CHANGED
@@ -145,17 +145,20 @@ def install_package_from_local_file(package_name, folder='packages'):
145
  def setup_environment():
146
  # Download required packages
147
  # 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
148
- packages_to_download = ['torch==1.13.1', 'torchvision==0.14.1', 'torchaudio==0.13.1']
149
- download_packages(packages_to_download, folder='packages/torch')
150
 
151
  # Install packages from local files
152
  install_package_from_local_file('torch', folder='packages/torch')
153
  install_package_from_local_file('torchvision', folder='packages/torch')
154
  install_package_from_local_file('torchaudio', folder='packages/torch')
155
-
156
- # subprocess.check_call([sys.executable, "-m", "pip", "install", "scikit-learn"])
157
- # subprocess.check_call([sys.executable, "-m", "pip", "install", "open3d"])
158
- # subprocess.check_call([sys.executable, "-m", "pip", "install", "easydict"])
 
 
 
159
 
160
  # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
161
  # if os.path.isdir(pc_util_path):
 
145
  def setup_environment():
146
  # Download required packages
147
  # 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
148
+ # packages_to_download = ['torch==1.13.1', 'torchvision==0.14.1', 'torchaudio==0.13.1']
149
+ # download_packages(packages_to_download, folder='packages/torch')
150
 
151
  # Install packages from local files
152
  install_package_from_local_file('torch', folder='packages/torch')
153
  install_package_from_local_file('torchvision', folder='packages/torch')
154
  install_package_from_local_file('torchaudio', folder='packages/torch')
155
+ install_package_from_local_file('scikit-learn', folder='packages/scikit-learn')
156
+ install_package_from_local_file('open3d', folder='packages/open3d')
157
+ install_package_from_local_file('easydict', folder='packages/easydict')
158
+
159
+ # download_packages(['scikit-learn'], folder='packages/scikit-learn')
160
+ # download_packages(['open3d'], folder='packages/open3d')
161
+ # download_packages(['easydict'], folder='packages/easydict')
162
 
163
  # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
164
  # if os.path.isdir(pc_util_path):