update packages
Browse files- .gitignore +0 -1
- 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 |
-
|
| 157 |
-
|
| 158 |
-
|
|
|
|
|
|
|
|
|
|
| 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):
|