what is the cudatoolkit you used to match the requirements.txt

THUDM / P-tuning

A novel method to tune language models. Codes and datasets for paper ``GPT understands, too''.

MIT License

923 stars 111 forks source link

what is the cudatoolkit you used to match the requirements.txt #29

Closed CSerxy closed 2 years ago

CSerxy commented 2 years ago

Hi authors,

I wonder if you could tell me how you launch the scripts/rte_pt_few_shot.sh? I met some difficulties in running that. Here is how I installed the packages:

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.0 -c pytorch

And it hints me that: AssertionError: Torch not compiled with CUDA enabled

Any help would be appreciated!

CSerxy commented 2 years ago

I solved this by rerun "pip install -r requirements.txt" with no need to find a specific cudatoolkit to install