![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 64
River-mao
这个作者很懒,什么都没留下…
展开
-
Pytorch使用torch.save的时候报out of memory
cuda runtime error (2) : out of memory at /pytorch/torch/csrc/generic/serialization.cpp:15应该是一个bug, 参考链接,参考评论,将模型转换到CPU之后,保存,再将模型转换到GPU即可以解决问题。代码如下: atten_t_net = atten_t_net.to(torch.device('cpu')) torch.save(atten_t_net.state_d原创 2021-01-05 12:43:48 · 797 阅读 · 0 评论 -
在jupyter lab中如何进入python虚拟环境
在jupyter lab中如何进入python虚拟环境打开终端新建一个名字为cv, python版本为3.7的虚拟环境,并进入conda create -n cv python=3.7conda activate cv在虚拟环境中安装lipykernelconda install ipykernel创新新的kernel# 最后一个参数为当前虚拟环境名ipython kernel install --user --name cv重启lab在notebook 中右上角更换kerne原创 2020-07-29 19:51:40 · 1948 阅读 · 0 评论 -
Visdom
Visdomvisdom是pytorch的可视化工具, 使用visdom方便跟踪网络训练时损失函数、正确率、学习率等参数的变化。1 下载安装python环境中应该要有pytorch.pip install visdom2 启动在终端输入:python -m visdom.server启动正常会出现:这里ctrl+右击网址,会在浏览器中打开一个窗口,如下:到这一步,整个visdom的运行环境已经完成。3 使用导入visdom模块:from visdom import Visd原创 2020-05-17 08:27:20 · 731 阅读 · 0 评论