python常用代码总结-python常用代码

常用代码片段及技巧

自动选择GPU和CPU

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

# model and tensor to device

vgg = models.vgg16().to(device)

切换当前目录

import os

try:

os.chdir(os.path.join(os.getcwd(), '..'))

print(os.getcwd())

except:

pass

打印模型参数

from torchsummary import summary

# 1 means in_channels

summary(model, (1, 28, 28))

将tensor的列表转换为tensor

x = torch.stack(tensor_list)

内存不够

Smaller batch size

torch.cuda.empty_cache()every few minibatches

debug tensor memory

resource` module is a Unix specific package as seen in https://docs.python.org/2/library/resource.html which is why it worked for you in Ubuntu, but raised an error when trying to use it in Windows.

Here is what solved it for me.

Downgrade to the Apache Spark 2.3.2 prebuild version

Install (or downgrade) jdk to version 1.8.0

My installed jdk was 1.9.0, which doesn't seem to be compatiable with spark 2.3.2 or 2.4.0

make sure that when you run java -version in cmd (command prompt), it show java version 8. If you are seeing version 9, you will need to change your system ENV PATH to ensure it points to java version 8.

Check this link to get help on changing the PATH if you have multiple java version installed.

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

def debug_memory():

import collections, gc, resource, torch

print('maxrss = {}'.format(

resource.getrusage(resource.RUSAGE_SELF).ru_maxrss))

tensors = collections.Counter((str(o.device), o.dtype, tuple(o.shape))

for o in gc.get_objects()

if torch.is_tensor(o))

for line in sorted(tensors.items()):

print('{} {}'.format(*line))

# example

import tensor

x = torch.tensor(3,3)

debug_memory()

y = torch.tensor(3,3)

debug_memory()

z = [torch.randn(i).long() for i in range(10)]

debug_memory()

python常用代码总结-python常用代码相关推荐

python实现中撤销上一步的代码mac_一个“MacBook”新手的Python“笨办法”自学之旅 #第六章：常用的简易Python命令、符号、代码、格式化字符串...
第六章:常用的简易Python命令.符号.代码.字符串 <"笨办法"学Python>这本书中,确实用了较多篇幅来介绍Python的一些常用简单的命令.符号.代码和字符串 ...
python常用代码大全-Python常用库大全，看看有没有你需要的
环境管理管理 Python 版本和环境的工具 p – 非常简单的交互式 python 版本管理工具. pyenv – 简单的 Python 版本管理工具. Vex – 可以在虚拟环境中执行命令. v ...
python常用代码大全-Python常用库大全及简要说明
环境管理管理 Python 版本和环境的工具 p:非常简单的交互式 python 版本管理工具.官网 pyenv:简单的 Python 版本管理工具.官网 Vex:可以在虚拟环境中执行命令.官网 v ...
python常用代码大全-Python常用库大全
Python常用库大全,看看有没有你需要的. 环境管理管理 Python 版本和环境的工具 p – 非常简单的交互式 python 版本管理工具. pyenv – 简单的 Python 版本管理工具 ...
vs2017常用设置显示行号格式化代码（Python）
vs2017常用设置显示行号格式化代码(Python) 第一步; 第二步: 搞定!!!!
python代码库-Python常用库大全
Python常用库大全,看看有没有你需要的. 环境管理管理 Python 版本和环境的工具 p – 非常简单的交互式 python 版本管理工具. pyenv – 简单的 Python 版本管理工具 ...
python1000个常用代码-介绍Python中几个常用的类方法
内置方法说明 __init__(self,...) 初始化对象,在创建新对象时调用 __del__(self) 释放对象,在对象被删除之前调用 __new__(cls,*args,**kwd) 实例 ...
Python培训教程：pycharm常用的快捷键合集
本期Python培训教程:pycharm常用的快捷键合集,希望能在后面的学习和工作中给大家带来一些帮助,首先我们来了解一下什么是PyCharm?PyCharm是一种Python IDE,它不仅具备了一 ...
干货 | 收藏！16段代码入门Python循环语句
(图片付费下载于视觉中国) 作者 | 李明江张良均周东平张尚佳,本文摘编自<Python3智能数据分析快速入门> 来源 | 大数据(ID:hzdashuju) [导读]本文将重点讲述 ...
值得收藏！16段代码入门Python循环语句
来源:大数据本文约6000字,建议阅读10分钟. 本文重点讲述for语句和while语句. [ 导读 ]本文重点讲述for语句和while语句.for语句属于遍历循环,while语句属于当型循环.除 ...

python常用代码总结-python常用代码

python常用代码总结-python常用代码相关推荐

最新文章

热门文章