Giter Site home page Giter Site logo

Comments (12)

MhLiao avatar MhLiao commented on August 16, 2024 2

@deepseek I think @ws-lin is right. Could you please try to use CUDA_VISIBLE_DEVICES=0 in the command.

from db.

ws-lin avatar ws-lin commented on August 16, 2024 1

I set it to work on just one gpu

from db.

 avatar commented on August 16, 2024 1

@liangkaihuan Thanks for the hint,
I found the solution, the issue was caused because I had CUDA Version 10.0.130, while @MhLiao tutorial had cudatoolkit=10.1.
In my case, the solution was to use cudatoolkit=10.0
Thanks all

from db.

ws-lin avatar ws-lin commented on August 16, 2024

the same error

from db.

llkn-2 avatar llkn-2 commented on August 16, 2024

I changed model.py(line:56) to
if self.training:
pred = self.model(data, training=self.training)
else:
pred = self.model(data)
and solved the problem.

from db.

 avatar commented on August 16, 2024

@ws-lin What code values did you change to solve this?
@liangkaihuan Upload your modified model.py here so i can test.

from db.

 avatar commented on August 16, 2024

@MhLiao CUDA_VISIBLE_DEVICES=0 allows to continue the process, but the end is "Segmentation fault", and in results folder nothing is recognized.

(db) home@home-desktop:~/p9/DB$ CUDA_VISIBLE_DEVICES=0 python eval.py experiments/seg_detector/totaltext_resnet18_deform_thre.yaml --resume ./modelz/totaltext_resnet18 --polygon --box_thresh 0.6
./datasets/total_text/
[INFO] [2019-12-05 20:11:06,534] Resuming from ./modelz/totaltext_resnet18
[INFO] [2019-12-05 20:11:06,563] Resumed from ./modelz/totaltext_resnet18
100%|██████████████████████████████████████████████████████████████████████████████████████| 300/300 [00:10<00:00, 27.78it/s]
[INFO] [2019-12-05 20:11:17,641] precision : 0.000000 (300)
[INFO] [2019-12-05 20:11:17,641] recall : 0.000000 (300)
[INFO] [2019-12-05 20:11:17,641] fmeasure : 0.000000 (1)
Segmentation fault

from db.

llkn-2 avatar llkn-2 commented on August 16, 2024

I tested CUDA_VISIBLE_DEVICES=0 without modifying code, and the forward() missing error can also be solved. I think Segmentation fault in your case may come from incorrect compilation of deformable convolution.

from db.

leedewdew avatar leedewdew commented on August 16, 2024

use CUDA_VISIBLE_DEVICES=0 can't always solve this problem when run the demo.
It still tells "TypeError: forward() missing 1 required positional argument: 'data'" .
Are there some other problems?

from db.

leedewdew avatar leedewdew commented on August 16, 2024

use CUDA_VISIBLE_DEVICES=0 can't always solve this problem when run the demo.
It still tells "TypeError: forward() missing 1 required positional argument: 'data'" .
Are there some other problems?

After hundreds of time trying, I find the issue is "CUDA_VISIBLE_DEVICES=0" doesn't take effect in my centos, while add one line of "os.environ["CUDA_VISIBLE_DEVICES"] = '3'" in demo.py finally saved me.

from db.

ygean avatar ygean commented on August 16, 2024

The problem is here: DataParallel doesn't properly handle kwargs

from db.

qiaoyaya2011 avatar qiaoyaya2011 commented on August 16, 2024

@deepseek it doesn't work. cuda version is equal cudatoolkit version.

from db.

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.