chn-lee-yumi / MaterialSearch

AI语义搜索本地素材。以图搜图、查找本地素材、根据文字描述匹配画面、视频帧搜索、根据画面描述搜索视频。Semantic search. Search local photos and videos through natural language.
GNU General Public License v3.0
799 stars 110 forks source link

运行install_torch_gpu.bat文件后,还是无法开启cuda加速 #81

Closed jianhudiyi closed 2 months ago

jianhudiyi commented 3 months ago

pip install -U torch --index-url=https://download.pytorch.org/whl/cu118 pause

up我在运行这行代码后,torch依旧没有安装成功,检查发现好像是安装到系统环境中去了,这个软件好像有个单独的环境(我也不是很懂程序这块,个人猜测),于是我查了下怎么在指定环境下安装包。以下是我的解决步骤: 1.我下载get-pip.py这个文件丢到了项目的根目录下。 2.然后再cmd中运行了这个python文件,python get-pip.py 3.用-m指定在该环境下安装torch,python -m pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

然后成功开启显卡加速了

chn-lee-yumi commented 3 months ago

你用的是整合包还是源码部署?

KENUNK commented 3 months ago

我也遇到同样的问题,按照这个哥们方法成功开启cuda加速,但是加速速度貌似不理想,gpu占用率0-5%左右,cuda占用率长期0%,10秒跳一次30%,显卡是nvdia 4070super 12G显存,也各种测试显卡环境cuda驱动都正常,可能某个参数可以更改是显卡使用率或者强度吧,不懂摸索中,感谢这位老哥的思路

chn-lee-yumi commented 3 months ago

加速速度貌似不理想,gpu占用率0-5%左右,cuda占用率长期0%,10秒跳一次30%

大概时间都花在了读取文件上面了吧,显卡不是瓶颈

jianhudiyi commented 3 months ago

你用的是整合包还是源码部署?

整合包

chn-lee-yumi commented 3 months ago

你用的是整合包还是源码部署?

整合包

整合包应该用CUDA加速安装.bat,而不是install_torch_gpu.bat,这个在使用说明里已经写了

jianhudiyi commented 3 months ago

你用的是整合包还是源码部署?

整合包

整合包应该用,而不是,这个在使用说明里已经写了CUDA加速安装.bat``install_torch_gpu.bat

也没有用 我还是安装不上torch,都运行了,我看了cuda加速器安装里的命令,是通过pip.pyz这个文件去安装torch,虽然文件里面是有torch的,但是会装到系统中去,可能是我系统中有多个python解释器,或许可以指定python解释器 例如:E:\MaterialSearchWindows\python.exe pip.pyz install torch --index-url=https://download.pytorch.org/whl/cu118 PAUSE 这样应该可以解决

chn-lee-yumi commented 3 months ago

或许可以指定python解释器 例如:E:\MaterialSearchWindows\python.exe pip.pyz install torch --index-url=https://download.pytorch.org/whl/cu118 PAUSE 这样应该可以解决

不知道你用的是哪个版本的整合包,20240609的版本已经加了相对路径,能保证用的是整合包内自带的python