Coder Social home page Coder Social logo

CUDA toolkit version about crfasrnn HOT 4 CLOSED

torrvision avatar torrvision commented on June 15, 2024
CUDA toolkit version

from crfasrnn.

Comments (4)

skeeph avatar skeeph commented on June 15, 2024 1

@Eromera thank you. I was doing it right. My mistake was that i didn't unomment lines
-gencode arch=compute_50,code=sm_50
-gencode arch=compute_50,code=compute_50
in Makefile.config. Because of this build script didn't generate code for new gpu architecture and i had error while running on gpu

from crfasrnn.

Eromera avatar Eromera commented on June 15, 2024

I am using CUDA 7.5 without trouble on a GTX Titan X. Is your GPU compatible? Did you correctly set the device id to use on gpu mode?

from crfasrnn.

skeeph avatar skeeph commented on June 15, 2024

@Eromera sorry for dumb questions, but could you give me instructions how to run python demo in gpu mode. I cloned this repo, compiled in gpu mode with cudnn, in python demo file I added strings
net.set_mode_gpu()
net.set_device(0)
But while performing net.predict([im]) I still got error:
F1104 21:28:48.601645 9549 im2col.cu:59] Check failed: error == cudaSuccess (8 vs. 0) invalid device function

What am I doing wrong?
P.S Sorry for my English

from crfasrnn.

Eromera avatar Eromera commented on June 15, 2024

To activate GPU mode you can just change the line in demo file:
net = caffe.Segmenter(MODEL_FILE, PRETRAINED)
With:
net = caffe.Segmenter(MODEL_FILE, PRETRAINED, True)

If you check the segmenter.py file in the caffe-crfrnn root you will find that the init function has that bool parameter and by default (false) it activates cpu mode unless you pass a True.

from crfasrnn.

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.