gpu
关注
分享
扫一扫
doublechenchenchen
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
DataParallel layers (multi GPU)
import torch.nn as nnimport torch.nn.parallelimport torch.backends.cudnn as cudnnmodel = nn.DataParallel(model)原创 2018-05-17 11:23:57 · 1592 阅读 · 0 评论 -
cudnn cuda version check
cuda : cat /usr/local/cuda/version.txtcudnn:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2原创 2018-05-16 21:32:54 · 623 阅读 · 0 评论