remove cuda setup
Browse files
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 |
-
|
| 219 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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'
|