Git Product home page Git Product logo

Comments (4)

slin000111 avatar slin000111 commented on August 18, 2024

可以尝试debug看看,是否有按is_gpu_available()指定device。极少数模型有这种情况。

from modelscope.

liuzhuangyuan avatar liuzhuangyuan commented on August 18, 2024

我菜狗,没找到问题原因,不过我通过一下方式可以切换为cpu,目前我的功能逻辑可以这么操作,不过我还是期望能在pipeline里面设置生效。

os.environ["CUDA_VISIBLE_DEVICES"] = "-1,"

import tensorflow as tf
# TensorFlow 配置:在某些情况下,TensorFlow 可能会使用自定义的 GPU 配置,覆盖了环境变量的设置。
# 可以尝试在 TensorFlow 导入之前设置 TensorFlow 的 GPU 配置。
# 设置 TensorFlow 的 GPU 配置
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_visible_devices(gpus, 'GPU')

from modelscope.

github-actions avatar github-actions commented on August 18, 2024

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.

from modelscope.

github-actions avatar github-actions commented on August 18, 2024

This issue was closed because it has been stalled for 5 days with no activity.

from modelscope.

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.