举个栗子:
requirements.txt:
protobuf
transformers==4.27.1
cpm_kernels
torch==2.2.2+cu121
gradio
mdtex2html
sentencepiece
accelerate
安装:
如下命令所示,后面添加 : -f https://download.pytorch.org/whl/torch_stable.html 即可!!!
pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html
这样就能很好的安装我们的torch gpu版本了!!完美!!!