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 · 1488 阅读 · 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 · 518 阅读 · 0 评论