Description
System Info
910B1,python==3.10,transformers==4.51.0
Who can help?
When torch==2.2.0 and torch-npu==2.2.0 and transformers>4.49, the error "AssertionError: Torch not compiled with CUDA enabled" will appear. <=4.49 has no problem. Upgrading to torch==2.4.0 and torch-npu==2.4.0 and transformers>4.49 allows training, but the graphics card cannot be specified. By default, it starts from card 0.
Information
- The official example scripts
- My own modified scripts
Tasks
- An officially supported task in the
examplesfolder (such as GLUE/SQuAD, ...) - My own task or dataset (give details below)
Reproduction
When torch==2.2.0 and torch-npu==2.2.0 and transformers>4.49, the error "AssertionError: Torch not compiled with CUDA enabled" will appear. <=4.49 has no problem. Upgrading to torch==2.4.0 and torch-npu==2.4.0 and transformers>4.49 allows training, but the graphics card cannot be specified. By default, it starts from card 0.
Expected behavior
When torch==2.2.0 and torch-npu==2.2.0 and transformers>4.49, the error "AssertionError: Torch not compiled with CUDA enabled" will appear. <=4.49 has no problem. Upgrading to torch==2.4.0 and torch-npu==2.4.0 and transformers>4.49 allows training, but the graphics card cannot be specified. By default, it starts from card 0.