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):
|