upate env
Browse files- .gitignore +2 -0
- 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 |
-
|
|
|
|
| 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):
|