Coder Social home page Coder Social logo

Comments (18)

Cloud-Liu avatar Cloud-Liu commented on July 19, 2024 1

Hello, I have the same problem. Have you solved it?

from malunet.

JCruan519 avatar JCruan519 commented on July 19, 2024

Hi! I am glad that you could follow our work.
In our "train.py" file, you could see "line 11". We set os.environ["CUDA_VISIBLE_DEVICES"] = "0", which means that our model will be run on the GPU of number 0. You could check whether your GPU number is correct.

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

Thank you for your reply. I have tried your method above, but it has no effect. I wonder if it's because my version of pytorch and torchvision is different from yours?

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

I have tried using both PC and server and I am very confused.

from malunet.

JCruan519 avatar JCruan519 commented on July 19, 2024

Maybe your pytorch and torchvision are installed as a cpu version. You can check the version in the terminal to ensure that you have installed the GPU version.

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

我一直都是GPU版本的torch,还是想不通是为什么

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

Have I always been a GPU torch, or can't figure out why

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

没解决 不知道怎么解决 试了很多办法

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

如果你解决了 可以教教我吗?

from malunet.

Cloud-Liu avatar Cloud-Liu commented on July 19, 2024

我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥

可以加个联系方式嘛? 这个代码对我很重要,希望可以探讨一下
QQ:1002367554

from malunet.

JCruan519 avatar JCruan519 commented on July 19, 2024

你好!我的代码是在win10环境下写的,因为这篇文章模型的体量较小,所以在Ubuntu或者服务器上并没有进行测试。我在上传代码之前用我本地三台不同的win10进行测试,都是可以使用GPU进行训练的。如果在win10下,还是无法使用GPU进行训练的话,我建议将本代码的模型部分拿出来,放到自己的工程文件中,修改参数然后进行训练。

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

你好!我的代码是在win10环境下写的,因为这篇文章模型的体量较小,所以在Ubuntu或者服务器上并没有进行测试。我在上传代码之前用我本地三台不同的win10进行测试,都是可以使用GPU进行训练的。如果在win10下,还是无法使用GPU进行训练的话,我建议将本代码的模型部分拿出来,放到自己的工程文件中,修改参数然后进行训练。

作者您好,你本地的win10系统都是什么型号的显卡?

from malunet.

JCruan519 avatar JCruan519 commented on July 19, 2024

这篇论文我实验时使用的是RTX3080,上传代码前还使用了A6000和RTX3090进行测试,都是win10下。

from malunet.

TWENTY-FOU avatar TWENTY-FOU commented on July 19, 2024

那请问您300epoch大概用了多长时间?

from malunet.

JCruan519 avatar JCruan519 commented on July 19, 2024

大约是6~7h吧,上限不会超过半天

from malunet.

icyclv avatar icyclv commented on July 19, 2024

感觉主要时间浪费在每轮重复resize了,在跑实验前把图像resize保存下比较好

from malunet.

guyuehhhhh avatar guyuehhhhh commented on July 19, 2024

我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥

可以加个联系方式嘛? 这个代码对我很重要,希望可以探讨一下 QQ:1002367554

我实验了几台电脑,下载这个代码复现都会每个epoch花费半个多小时,到底哪里错了啊,谁能救救我

from malunet.

Related Issues (9)

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.