You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Describe the bug
After import open3d, taichi can't initialize multiple cuda devices. To Reproduce
importtaichiastiimportosfrommultiprocessingimportProcessimportopen3d#no bug if comment this line@ti.kerneldefdo_something():
foriinrange(100000000):
a=i**2defrun_taichi_on_gpu(gpu_id):
# Set the CUDA_VISIBLE_DEVICES environment variable to the desired GPUos.environ["CUDA_VISIBLE_DEVICES"] =str(gpu_id)
# Initialize Taichi to run on the specified GPUti.init(arch=ti.cuda)
# Run the Taichi kerneldo_something()
if__name__=="__main__":
# List of GPUs you want to usegpus= [0,1,2] # Example for using GPU 0 and GPU 1# Create a separate process for each GPUprocesses= [Process(target=run_taichi_on_gpu, args=(gpu_id,)) forgpu_idingpus]
# Start all processesforpinprocesses:
p.start()
# Wait for all processes to finishforpinprocesses:
p.join()
Log/Screenshots
RuntimeError: [cuda_driver.h:operator()@92] CUDA Error CUDA_ERROR_NOT_INITIALIZED: initialization error while calling init (cuInit)
Additional comments
If possible, please also consider attaching the output of command ti diagnose. This produces the detailed environment information and hopefully helps us diagnose faster.
If you have local commits (e.g. compile fixes before you reproduce the bug), please make sure you first make a PR to fix the build errors and then report the bug.
The text was updated successfully, but these errors were encountered:
Describe the bug
After
import open3d
, taichi can't initialize multiple cuda devices.To Reproduce
Log/Screenshots
RuntimeError: [cuda_driver.h:operator()@92] CUDA Error CUDA_ERROR_NOT_INITIALIZED: initialization error while calling init (cuInit)
Additional comments
If possible, please also consider attaching the output of command
ti diagnose
. This produces the detailed environment information and hopefully helps us diagnose faster.If you have local commits (e.g. compile fixes before you reproduce the bug), please make sure you first make a PR to fix the build errors and then report the bug.
The text was updated successfully, but these errors were encountered: