colin1842 commited on
Commit
d8f7bb3
·
1 Parent(s): 66266ad

remove cuda setup

Browse files
Files changed (1) hide show
  1. script.py +70 -6
script.py CHANGED
@@ -174,6 +174,15 @@ def install_package_from_local_file(package_name, folder='packages'):
174
  except subprocess.CalledProcessError as e:
175
  print(f"Failed to install package from {pth}. Error: {e}")
176
 
 
 
 
 
 
 
 
 
 
177
  def setup_environment():
178
  # Uninstall torch if it is already installed
179
  # packages_to_uninstall = ['torch', 'torchvision', 'torchaudio']
@@ -202,21 +211,76 @@ def setup_environment():
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')
206
- install_package_from_local_file('setuptools', folder='packages')
 
207
  # download_packages(['scikit-learn'], folder='packages/scikit-learn')
208
  # download_packages(['open3d'], folder='packages/open3d')
209
  # download_packages(['easydict'], folder='packages/easydict')
210
 
211
- setup_cuda_environment()
212
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
213
  pc_util_path = os.path.join(os.getcwd(), 'pc_util')
214
  if os.path.isdir(pc_util_path):
215
  os.chdir(pc_util_path)
216
- subprocess.check_call([sys.executable, "setup.py", "install"])
217
  os.chdir("..")
218
- else:
219
- print(f"Directory {pc_util_path} does not exist")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
220
 
221
  def setup_cuda_environment():
222
  cuda_home = '/usr/local/cuda'
 
174
  except subprocess.CalledProcessError as e:
175
  print(f"Failed to install package from {pth}. Error: {e}")
176
 
177
+ def install_which():
178
+ try:
179
+ # Attempt to install which if it's not available
180
+ subprocess.check_call(['sudo', 'apt-get', 'install', '-y', 'which'])
181
+ print("Which installed successfully.")
182
+ except subprocess.CalledProcessError as e:
183
+ print(f"An error occurred while installing which: {e}")
184
+ sys.exit(1)
185
+
186
  def setup_environment():
187
  # Uninstall torch if it is already installed
188
  # packages_to_uninstall = ['torch', 'torchvision', 'torchaudio']
 
211
  # install_package_from_local_file('packages/torch/torchaudio-2.1.0-cp38-cp38-manylinux1_x86_64.whl', folder='packages/torch')
212
  # install_package_from_local_file('scikit-learn', folder='packages')
213
  # install_package_from_local_file('open3d', folder='packages')
214
+ # install_package_from_local_file('easydict', folder='packages')
215
+ # install_package_from_local_file('setuptools', folder='packages')
216
+ # install_package_from_local_file('ninja', folder='packages')
217
  # download_packages(['scikit-learn'], folder='packages/scikit-learn')
218
  # download_packages(['open3d'], folder='packages/open3d')
219
  # download_packages(['easydict'], folder='packages/easydict')
220
 
 
221
 
222
+
223
+ # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
224
+ # if os.path.isdir(pc_util_path):
225
+ # os.chdir(pc_util_path)
226
+ # subprocess.check_call([sys.executable, "setup.py", "install"])
227
+ # os.chdir("..")
228
+ # else:
229
+ # print(f"Directory {pc_util_path} does not exist")
230
+
231
+ # try:
232
+ # subprocess.check_call(['which', 'which'])
233
+ # except subprocess.CalledProcessError:
234
+ # Ensure 'which' is installed
235
+ # try:
236
+ # subprocess.check_call(['which', 'which'])
237
+ # except subprocess.CalledProcessError:
238
+ # install_which()
239
+
240
+ # install_package_from_local_file('easydict', folder='packages')
241
+ # install_package_from_local_file('setuptools', folder='packages')
242
+ # install_package_from_local_file('ninja', folder='packages')
243
+ # install_package_from_local_file('torch', folder='packages/torch')
244
+ # install_package_from_local_file('torchvision', folder='packages/torchvision')
245
+ # install_package_from_local_file('torchaudio', folder='packages/torchaudio')
246
+
247
  pc_util_path = os.path.join(os.getcwd(), 'pc_util')
248
  if os.path.isdir(pc_util_path):
249
  os.chdir(pc_util_path)
250
+ subprocess.check_call([sys.executable, "setup.py", "install"], cwd=pc_util_path)
251
  os.chdir("..")
252
+
253
+ # pc_util_path = os.path.join(os.getcwd(), 'pc_util')
254
+ # if os.path.isdir(pc_util_path):
255
+ # os.chdir(pc_util_path)
256
+ # try:
257
+ # # # install_package_from_local_file('build', folder='packages/build')
258
+ # # # subprocess.check_call([sys.executable, "-m", "build", "--wheel"])
259
+ # # # subprocess.check_call([sys.executable, "-m", "pip", "install", "dist/pc_util-1.0-py3.8-linux-x86_64.egg"])
260
+ # # subprocess.check_call([sys.executable, "setup.py", "install", "--single-version-externally-managed", "--record", "record.txt"])
261
+ # # subprocess.check_call([sys.executable, "setup.py", "install"], cwd=pc_util_path)
262
+ # # subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-index", "--find-links", "../packages/build", "build"])
263
+ # # subprocess.check_call([sys.executable, "-m", "build", "--wheel"], cwd=pc_util_path)
264
+ # # egg_file = next(f for f in os.listdir(os.path.join(pc_util_path, "dist")) if f.endswith(".egg"))
265
+ # # subprocess.check_call([sys.executable, "-m", "pip", "install", os.path.join(pc_util_path, "dist", egg_file)])
266
+ # # Temporarily add a fake 'which' command to the PATH
267
+ # # with open("/tmp/which", "w") as f:
268
+ # # f.write("#!/bin/bash\necho /usr/bin/$1")
269
+ # # os.chmod("/tmp/which", 0o755)
270
+ # # os.environ["PATH"] = "/tmp:" + os.environ["PATH"]
271
+
272
+ # # subprocess.check_call([sys.executable, "setup.py", "install"], cwd=pc_util_path)
273
+ # except subprocess.CalledProcessError as e:
274
+ # print(f"Failed to build and install pc_util. Error: {e}")
275
+ # subprocess.check_call([sys.executable, "setup.py", "install"], cwd=pc_util_path)
276
+ # subprocess.check_call([sys.executable, "-m", "build"])
277
+ # subprocess.check_call([sys.executable, "-m", "pip", "install", "."])
278
+ # else:
279
+ # print(f"Directory {pc_util_path} does not exist")
280
+ # subprocess.run(['cd', 'pc_util'], shell=True)
281
+
282
+ # # Install the package
283
+ # subprocess.run(['python', 'setup.py', 'install'], cwd='pc_util', shell=True)
284
 
285
  def setup_cuda_environment():
286
  cuda_home = '/usr/local/cuda'