Giter VIP home page Giter VIP logo

Comments (10)

yangheng95 avatar yangheng95 commented on May 21, 2024

您能贴出代码错误的trace吗

from pyabsa.

SongShihao731 avatar SongShihao731 commented on May 21, 2024

image

from pyabsa.

yangheng95 avatar yangheng95 commented on May 21, 2024

我注意到这个问题了,autocuda的是采用的GitHub上的开源代码,是通过系统命令nvidia-smi查询cuda,可能你的命令配置有问题。能让我看一下你的环境中该命令的输出吗

from pyabsa.

SongShihao731 avatar SongShihao731 commented on May 21, 2024

image

from pyabsa.

yangheng95 avatar yangheng95 commented on May 21, 2024

这个输出确实不大一样,不清楚是否是新版本导致的问题。我发布一个补丁,让您可以直接手动指定cuda device,可以吗?
如果确实是nvidia查询结果改版的话,我后面会修改autocuda源码

from pyabsa.

SongShihao731 avatar SongShihao731 commented on May 21, 2024

麻烦您了

from pyabsa.

yangheng95 avatar yangheng95 commented on May 21, 2024

我发不了1.0.7.1版本,宁可以使用auto_deivce='cuda'指定GPU,请反馈一下测试结果,谢谢!

from pyabsa.

SongShihao731 avatar SongShihao731 commented on May 21, 2024

代码显示可以正常运行了,感谢您
image

from pyabsa.

yangheng95 avatar yangheng95 commented on May 21, 2024

除此之外您还可以加载完模型后手动切换device: https://github.com/yangheng95/PyABSA/blob/release/examples/common_usages/change_loading_device.py

from pyabsa.

SongShihao731 avatar SongShihao731 commented on May 21, 2024

谢谢~

from pyabsa.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.