Comments (4)
Thank you!
I mean when I want to terminate the training, I would use Ctrl+D in the terminal. Sometimes one process would still be there occupying the GPU. Then I have to kill the process by its ID.
from pytorch-distributed.
Of course, if you do torch.cuda.set_device(local_rank)
it is okay to use model.cuda()
instead of model.cuda(args.gpu)
. The official repo kept the old fashion of best practice based on previous versions.
And what do u mean by 'cannot kill all the processes by Ctrl+D'? Everything goes well for me.
from pytorch-distributed.
Thank you!
I mean when I want to terminate the training, I would use Ctrl+D in the terminal. Sometimes one process would still be there occupying the GPU. Then I have to kill the process by its ID.
Me too... Do you have any advice?
from pytorch-distributed.
I finally encountered a situation where the GPU memory was not cleared after using ctrl+c to kill the process today. It happened when I used a custom CUDA module. If you encounter the same situation, maybe you can refer to facebookresearch/fairseq#487, which works for me (even though does not look elegant).
I will try to find the cause of this problem, but it seems not easy to locate...
from pytorch-distributed.
Related Issues (20)
- Does the test accuracy need to be synchronized in distributed.py? HOT 7
- 大佬,请问如何指定gpu训练
- apex 并行加速中的data_prefetcher normalize为什么做两次? HOT 1
- 您好,我想请问一下Pytorch 在多进程分布式训练的时候,一开始加载数据集dataloader的时候,由于是多进程的,就会导致多个进程一起加载dataloader,这些dataloder全部加载到内存中使得内存爆炸,请问有没有什么办法解决呢
- 多卡显存占用不均衡 HOT 1
- No need to divide with nprocs again HOT 1
- Windows 提示Distributed package doesn't have NCCL "Distributed package doesn't have NCCL built in HOT 1
- how can i modify it
- how can i modify it ?
- RuntimeError: Timed out initializing process group in store based barrier on rank: 2, for key: store_based_barrier_key:1 (world_size=2, worker_count=4, timeout=0:30:00) HOT 2
- 2080Ti多卡训练,GPU利用率忽高忽低,这个怎么回事 HOT 2
- Can't pickle <function main_worker
- Connections time out
- DDP掉精度 HOT 3
- distributed.py seed HOT 6
- About saving model in the multi-GPU training. HOT 3
- 使用mp.spawn启动比torch.distributed.launch慢很多 HOT 2
- 关于损失backward问题 HOT 1
- 请问使用DistributedSampler,各个GPU的数据是如何分配的?是连续(互不相同)的还是相同的? HOT 1
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 pytorch-distributed.