Pytorch
sean_xyz
问渠那得清如许,为有源头活水来
展开
-
深度学习工程实践Tips(长期更新)
1. 训练阶段加速 * 使用混合精度加速训练 推荐使用NVIDIA的APEX进行训练加速,在2060S显卡上,大约可以获得一倍的综合速度提升(显存使用减少,可以适当增加batchsize, 提高速度)。精度下降几乎察觉不到。APEX保存和加载的模型参数仍然是float32类型的,所以不用担心预训练模型重用的问题。使用起来也十分简单:h...原创 2019-12-13 11:09:22 · 414 阅读 · 0 评论 -
Ubuntu 18.04下 安装pytorch
1. 安装CUDA 包括显卡驱动,CUDA, CUDNN 请参考博客 https://mp.csdn.net/postedit/84586319 2. 安装 Anaconda 使用命令行下载安装文件,并安装curl -O https://repo.anaconda.com/archive/Anaconda3-5.2.0-...原创 2018-11-28 17:16:05 · 3048 阅读 · 0 评论