安装gpu版本的巨坑:
不要用清华源安装,安装的是cpu版本的
pip install torch==1.11.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
这个指令不可以的。
这里提供一个方法
进入镜像站
https://download.pytorch.org/whl/torch/
选择需要whl文件下载之后,确定好文件的目录和名字后,在conda的虚拟环境下执行下面的指令即可
pip3 install D:\anaconda3\envs\llama_factory\torch-2.3.1+cu121-cp310-cp310-win_amd64.whl
大家根据自己的文件名和文件目录修改这条指令。