- 博客(7)
- 收藏
- 关注
原创 docker内torch.cuda.is_available()=FALSE
报错为Error 500: named symbol not found (Triggered internally at /opt/conda/conda-记录docker内torch.cuda.is_available为false的解决。原因可能为更新了显卡驱动,导致显卡驱动与cuda版本不兼容,解决方案就是回退驱动版本,在设备管理器中。
2024-07-23 16:16:34 104 1
转载 基于pytorch得预训练模型
Resnet:model_urls = {undefined 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e35...
2022-01-14 17:27:11 150
原创 Pyinstaller 打包pytorch时遇到得问题
在打包pytorch项目时遇到了torch打包不进去得问题(好像是),exe打开之后闪退,报错如下仔细查看打包时得打印信息,发现有一句lib not found torch_python.dll *******,考虑是这句话得原因,但是查自己打包得路径下是有这个dll库得,会不会是没有找到自己环境中得这个dll,最后通过在打包时把这个路径链接进来,解决了问题!!具体命令为 pyinstaller --paths 你没找到得dll路径,比如我得在**torch/lib下 ***.py...
2021-11-30 11:38:25 2527 3
原创 RuntimeError: Trying to backward through the graph a second time, but the buffers have already been
遇到的问题记录:这个错误的意思是在进行反向传播时,前面缓冲区里的计算图已经被释放了。在前向传播过程中每进行一次计算都会被加入到缓冲区的计算图中,而反向传播时,根据计算图中需要进行梯度传递的部分进行反向梯度计算。我遇到的这个问题是在同时迭代更新两个网络的过程。当第一个网络进行backward之后,相应的计算图就已经被释放了,而两个网络的训练损失有部分重叠,所以当第二个网络backward时,就会出现这个问题。解决的方法是将重叠部分的损失进行重新计算用于第二个网络。注意两个网络的损失要分开计算,在计算过
2021-10-05 15:52:26 265
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人