--num-gpus 2 error

#26
by zjtzpanxb - opened

I have 2*4090,I add the parameter --num-gpus 2 run, The prompt error is as follows:
aa.png

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

aa.png

![bb.png]({"error":"Only PNG, JPG, JPEG, GIF, MP4, MOV, QT, WEBM, MP3, MPGA, WAV files are supported"})

cc.png

Sign up or log in to comment