I have met the problem when i run the"pip install -e ".[train]"":
No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4'
Traceback (most recent call last):
File "", line 2, in
File "", line 34, in
File "C:\Users\chenminjie\AppData\Local\Temp\pip-install-gc47w0br\deepspeed_4327a8fe39eb48f8954773a0d84e7b45\setup.py", line 163, in
abort(f"Unable to pre-compile {op_name}")
File "C:\Users\chenminjie\AppData\Local\Temp\pip-install-gc47w0br\deepspeed_4327a8fe39eb48f8954773a0d84e7b45\setup.py", line 51, in abort
assert False, msg
AssertionError: Unable to pre-compile async_io
[WARNING] Torch did not find cuda available, if cross-compiling or running with cpu only you can ignore this message. Adding compute capability for Pascal, Volta, and Turing (compute capabilities 6.0, 6.1, 6.2)
I have met the problem when i run the"pip install -e ".[train]"":
No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4' Traceback (most recent call last): File "", line 2, in
File "", line 34, in
File "C:\Users\chenminjie\AppData\Local\Temp\pip-install-gc47w0br\deepspeed_4327a8fe39eb48f8954773a0d84e7b45\setup.py", line 163, in
abort(f"Unable to pre-compile {op_name}")
File "C:\Users\chenminjie\AppData\Local\Temp\pip-install-gc47w0br\deepspeed_4327a8fe39eb48f8954773a0d84e7b45\setup.py", line 51, in abort
assert False, msg
AssertionError: Unable to pre-compile async_io
[WARNING] Torch did not find cuda available, if cross-compiling or running with cpu only you can ignore this message. Adding compute capability for Pascal, Volta, and Turing (compute capabilities 6.0, 6.1, 6.2)