pytorch
先出个点金手
这个作者很懒,什么都没留下…
展开
-
Mac 配置pytorch
1. anaconda环境配置最近刚把bash 升级为zsh,所以很多环境变量要重新配置在vim中打开~/.zshrc 加入保存退出2.安装pytorchconda create -n pytorch python=3.6根据需要配置不同的python版本source activate pytorch在进行安装pip3 install torch torchvision...原创 2018-07-15 18:03:19 · 1129 阅读 · 0 评论 -
pytorch 减小显存消耗,优化显存使用,避免out of memory
本文是整理了大神的两篇博客:如何计算模型以及中间变量的显存占用大小:https://oldpan.me/archives/how-to-calculate-gpu-memory如何在Pytorch中精细化利用显存:https://oldpan.me/archives/how...转载 2018-08-16 17:16:18 · 2210 阅读 · 0 评论