PyTorch笔记

2019-01-05 34

指定GPU

先用nvidia-smi查看gpu负载状态,然后跑torch的时候,指定可见的gpu ID就行

在运行python blablabla前面加一句变成CUDA_VISIBLE_DEVICES=0,1,2 python blablabla,意义是可见的gpu有0,1,2号,而且torch会按照0,1,2这个顺序来使用gpu

optimizer和cuda的bug

大概出现这种错时

加上类似这份代码即可

Github issue

欢迎留言>_<

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据