bob_chen_csdn的博客

pytorch GPU 计算过程中出现内存耗尽 Pytorch GPU运算过程中会出现：“cuda runtime error(2): out of memory”这样的错误。通常，这种错误是由于在循环中使用全局变量当做累加器，且累加梯度信息的缘故，用官方的说法就是："accumulate history across your training loop"。在默认情况下，开启梯度计算的Tenso...

Pytorch 模型存储 pytorch 模型存储有两种方法： 存储参数 #存储 torch.save(the_model.state_dict(),PATH) #恢复 the_model=THe_model() the_model.load_state_dict(torch.load(PATH)) 存储整个模型 #存储 torch.save(the_model,PATH) #恢复 the_mo...

Ubuntu16.04+NVIDIA TITAN V 驱动+CUDA8.0安装 1.Ubuntu16.4安装： 直接在ubuntu官网下载Ubuntu镜像，一步步安装 2.安装NVIDIA驱动： 1.https://www.geforce.cn/drivers在这个链接里寻找版本为390.87的驱动，并下载。 2.开机，在终端输入如下命令，将旧的驱动禁止掉 sudo chmod 666 /etc/...

Pythorch 数据初始化 pytorch在数据从原始数据集里面获取以后（一般处理成numpy数组），需要以下步骤： 1.构造DataClass torch.utils.data.Dataset是一个表示数据的抽象类，在构造自己的数据集时候，首先应该构造一个他的子类，并且该子类需要重写下面两个方法： __len__，这个方法在使用len(Dataset)时候会被调动，用于返回数据集Datase...

pytorch 动态网络+权值共享 pytorch以动态图著称，下面以一个栗子来实现动态网络和权值共享技术: # -*- coding: utf-8 -*- import random import torch class DynamicNet(torch.nn.Module): def __init__(self, D_in, H, D_out): """ ...

NVIDIA 2080Ti+NVIDIA Titan V 深度学习服务器配置安装过程