- 博客(7)
- 收藏
- 关注
RSS订阅原创 NVIDIA 2080Ti + NVIDIA Titan V 深度学习服务器安装配置过程
NVIDIA 2080Ti+NVIDIA Titan V 深度学习服务器配置安装过程
2019-07-13 17:15:53
4601
原创 pytorch动态网络以及权重共享
pytorch 动态网络+权值共享pytorch以动态图著称，下面以一个栗子来实现动态网络和权值共享技术:# -*- coding: utf-8 -*-import randomimport torchclass DynamicNet(torch.nn.Module): def __init__(self, D_in, H, D_out): """ ...
2018-11-02 20:29:27
7035
原创 pytorch数据导入以及预处理
Pythorch 数据初始化pytorch在数据从原始数据集里面获取以后（一般处理成numpy数组），需要以下步骤：1.构造DataClasstorch.utils.data.Dataset是一个表示数据的抽象类，在构造自己的数据集时候，首先应该构造一个他的子类，并且该子类需要重写下面两个方法：__len__，这个方法在使用len(Dataset)时候会被调动，用于返回数据集Datase...
2018-11-01 17:32:57
6129
原创 pytorch 存储模型
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...
2018-10-29 22:51:46
446
原创 pytorch GPU内存耗尽
pytorch GPU 计算过程中出现内存耗尽Pytorch GPU运算过程中会出现：“cuda runtime error(2): out of memory”这样的错误。通常，这种错误是由于在循环中使用全局变量当做累加器，且累加梯度信息的缘故，用官方的说法就是："accumulate history across your training loop"。在默认情况下，开启梯度计算的Tenso...
2018-10-29 22:01:04
14674
原创 ubuntu16
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/...
2018-10-24 11:56:32
245
转载 Markdown
欢迎使用Markdown编辑器写博客本Markdown编辑器使用StackEdit修改而来，用它写博客，将会带来全新的体验哦：Markdown和扩展Markdown简洁的语法代码块高亮图片链接和图片上传LaTex数学公式UML序列图和流程图离线写博客导入导出Markdown文件丰富的快捷键快捷键加粗 Ctrl + B 斜体 Ctrl + I...
2018-08-05 17:54:28
105
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人