pytorch
AliceWulawula
这个作者很懒,什么都没留下…
展开
-
baichuan 量化版本模型在windows下的加载失败问题
从中选择当前需要装的版本替换即可。原创 2023-11-06 19:51:28 · 72 阅读 · 1 评论 -
使用pip安装库时,自动卸载pytorch gpu版本并安装pytorch cpu版本的问题
使用pip安装库时,有些库的安装会卸载掉原本安装好的pytorch-gpu版本,并安装了pytorch-cpu版本,导致无法使用gpu,比如xformers,transformers库等。这导致配环境的时候非常麻烦,并且容易报废环境。一般的方案是重新配一个环境避免冲突,但是如果基于requirements.txt安装的环境在安装好了pytorch很可能无法使用gpu。因此考虑修复环境。目前想到了两个方案。原创 2023-11-06 19:48:08 · 464 阅读 · 1 评论