colin1842 commited on
Commit
cac83f4
·
1 Parent(s): e0581cb

remove torch install

Browse files
Files changed (1) hide show
  1. script.py +6 -6
script.py CHANGED
@@ -176,9 +176,9 @@ def install_package_from_local_file(package_name, folder='packages'):
176
 
177
  def setup_environment():
178
  # Uninstall torch if it is already installed
179
- packages_to_uninstall = ['torch', 'torchvision', 'torchaudio']
180
- for package in packages_to_uninstall:
181
- uninstall_package(package)
182
  # Download required packages
183
  # 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
184
  # pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
@@ -197,9 +197,9 @@ def setup_environment():
197
  # download_packages(packages_to_download, download_folder)
198
 
199
  # Install packages from local files
200
- install_package_from_local_file('torch', folder='packages')
201
- install_package_from_local_file('packages/torch/torchvision-0.16.0-cp38-cp38-manylinux1_x86_64.whl', folder='packages/torch')
202
- install_package_from_local_file('packages/torch/torchaudio-2.1.0-cp38-cp38-manylinux1_x86_64.whl', folder='packages/torch')
203
  # install_package_from_local_file('scikit-learn', folder='packages')
204
  # install_package_from_local_file('open3d', folder='packages')
205
  install_package_from_local_file('easydict', folder='packages')
 
176
 
177
  def setup_environment():
178
  # Uninstall torch if it is already installed
179
+ # packages_to_uninstall = ['torch', 'torchvision', 'torchaudio']
180
+ # for package in packages_to_uninstall:
181
+ # uninstall_package(package)
182
  # Download required packages
183
  # 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
184
  # pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
 
197
  # download_packages(packages_to_download, download_folder)
198
 
199
  # Install packages from local files
200
+ # install_package_from_local_file('torch', folder='packages')
201
+ # install_package_from_local_file('packages/torch/torchvision-0.16.0-cp38-cp38-manylinux1_x86_64.whl', folder='packages/torch')
202
+ # install_package_from_local_file('packages/torch/torchaudio-2.1.0-cp38-cp38-manylinux1_x86_64.whl', folder='packages/torch')
203
  # install_package_from_local_file('scikit-learn', folder='packages')
204
  # install_package_from_local_file('open3d', folder='packages')
205
  install_package_from_local_file('easydict', folder='packages')