cbensimon HF staff commited on
Commit
04a2147
1 Parent(s): 0dd1fb3

Create open3d_zerogpu_fix.py

Browse files
Files changed (1) hide show
  1. open3d_zerogpu_fix.py +7 -0
open3d_zerogpu_fix.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import fileinput
2
+ import site
3
+ from pathlib import Path
4
+
5
+ with fileinput.FileInput(f'{site.getsitepackages()[0]}/open3d/__init__.py', inplace=True) as file:
6
+ for line in file:
7
+ print(line.replace('_pybind_cuda.open3d_core_cuda_device_count()', '1'), end='')