![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Deep learning
FakeGhost01
行百里者半九十。
展开
-
pytorch 的train和eval的极简描述
训练需要 model.train() :启用 BatchNormalization 和 Dropout 测试必要 (pytorch是默认开启BatchNormalization 和 Dropout的,测试或者pre-train的时候必须使用) model.eval() :不启用 BatchNormalization 和 Dropout ...原创 2020-03-14 22:40:37 · 251 阅读 · 0 评论 -
已有CUDA情况下安装cuDNN
1 查询cuDNN 安装情况 cuDNN是GPU加速计算深层神经网络的库。 突然发现cudnn没安装上: $ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 得到: /usr/local/cuda/include/cudnn.h 为了查证是否是自己记错目录,查询一下 $ find -name cudnn.h 发现确实没有安装好...原创 2019-11-20 00:34:03 · 541 阅读 · 0 评论