Comments (18)
Hello, I have the same problem. Have you solved it?
from malunet.
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.
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.
I have tried using both PC and server and I am very confused.
from malunet.
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.
我一直都是GPU版本的torch,还是想不通是为什么
from malunet.
Have I always been a GPU torch, or can't figure out why
from malunet.
没解决 不知道怎么解决 试了很多办法
from malunet.
如果你解决了 可以教教我吗?
from malunet.
我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥
from malunet.
我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥
可以加个联系方式嘛? 这个代码对我很重要,希望可以探讨一下
QQ:1002367554
from malunet.
你好!我的代码是在win10环境下写的,因为这篇文章模型的体量较小,所以在Ubuntu或者服务器上并没有进行测试。我在上传代码之前用我本地三台不同的win10进行测试,都是可以使用GPU进行训练的。如果在win10下,还是无法使用GPU进行训练的话,我建议将本代码的模型部分拿出来,放到自己的工程文件中,修改参数然后进行训练。
from malunet.
你好!我的代码是在win10环境下写的,因为这篇文章模型的体量较小,所以在Ubuntu或者服务器上并没有进行测试。我在上传代码之前用我本地三台不同的win10进行测试,都是可以使用GPU进行训练的。如果在win10下,还是无法使用GPU进行训练的话,我建议将本代码的模型部分拿出来,放到自己的工程文件中,修改参数然后进行训练。
作者您好,你本地的win10系统都是什么型号的显卡?
from malunet.
这篇论文我实验时使用的是RTX3080,上传代码前还使用了A6000和RTX3090进行测试,都是win10下。
from malunet.
那请问您300epoch大概用了多长时间?
from malunet.
大约是6~7h吧,上限不会超过半天
from malunet.
感觉主要时间浪费在每轮重复resize了,在跑实验前把图像resize保存下比较好
from malunet.
我也没解决,我把那行用多GPU的代码注释了,设置了numworkers=16,但是跑的时候显存利用率还是0%,不知道为啥
可以加个联系方式嘛? 这个代码对我很重要,希望可以探讨一下 QQ:1002367554
我实验了几台电脑,下载这个代码复现都会每个epoch花费半个多小时,到底哪里错了啊,谁能救救我
from malunet.
Related Issues (9)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from malunet.