pytorch 实现删除tensor中的指定行列
yipeiwu_com6年前
前言 在pytorch中， 想删除tensor中的指定行列，原本以为有个函数或者直接把某一行赋值为[]就可以，结果发现没这么简单，因此用了一个曲线救国方法，希望如果有更直接的方法，请大家...
Pytorch释放显存占用方式
yipeiwu_com6年前
如果在python内调用pytorch有可能显存和GPU占用不会被自动释放，此时需要加入如下代码 torch.cuda.empty_cache() 我们来看一下官方文档的说明 Relea...
Python hashlib常见摘要算法详解
yipeiwu_com6年前
这篇文章主要介绍了Python hashlib常见摘要算法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 Python的hashl...
python批量处理txt文件的实例代码
yipeiwu_com6年前
通过python对多个txt文件进行处理 读取路径，读取文件 获取文件名，路径名 对响应的文件夹名字进行排序 对txt文件内部的数据相应的某一列/某一行进行均值处理...
PyTorch 随机数生成占用 CPU 过高的解决方法
yipeiwu_com6年前
PyTorch 随机数生成占用 CPU 过高的问题 今天在使用 pytorch 的过程中，发现 CPU 占用率过高。经过检查，发现是因为先在 CPU 中生成了随机数，然后再调用.to(d...
详解python破解zip文件密码的方法
yipeiwu_com6年前
1、单线程破解纯数字密码 注意： 不包括数字0开头的密码 import zipfile,time,sys start_time = time.time() def extract()...
Python如何读取文件中图片格式
yipeiwu_com6年前
这篇文章主要介绍了Python如何读取文件中图片格式,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 正常情况下，图片命名的后缀和图片的...
PyTorch使用cpu加载模型运算方式
yipeiwu_com6年前
没gpu没cuda支持的时候加载模型到cpu上计算 将 model = torch.load(path, map_location=lambda storage, loc: stor...
Python跑循环时内存泄露的解决方法
yipeiwu_com6年前
Python跑循环时内存泄露 今天在用Tensorflow跑回归做测试时，仅仅需要循环四千多次 （补充说一句，我在个人PC上跑的）。运行以后，我就吃饭去了。等我回来后，Console窗口...
pytorch程序异常后删除占用的显存操作
yipeiwu_com6年前
1-删除模型变量 del model_define 2-清空CUDA cache torch.cuda.empty_cache() 3-步骤2（异步）需要一定时间，设置时延...
