草庐IT

torch-geometric

全部标签

torch.hub.load 加载本地模型(已解决)

背景运行网上的项目,有时会卡住或者超时,原因是torch.hub.load默认会去网上找模型,有时会因为网络问题而报错解决方法不让torch.hub.load联网下载模型,改为torch.hub.load加载本地模型。本地模型默认的下载路径是:/root/.cache/torch/hub……原始:vad_model,funcs=torch.hub.load(repo_or_dir="snakers4/silero-vad",model="silero_vad",trust_repo=True)改为:vad_model,funcs=torch.hub.load(repo_or_dir="/roo

【pytorch函数笔记】torch.sum()、torch.unsqueeze()

1.torch.sumtorch.sum(imgs,dim=0)#按列求和torch.sum(imgs,dim=1) #按行求和imgs=torch.Tensor([iforiinzip(range(10),range(10))])print(imgs)s1=torch.sum(imgs,dim=0)#按列求和s2=torch.sum(imgs,dim=1)#按行求和print(s1)print(s2)2.torch.unsqueeze将每个图像张量的维度扩展,即在每个图像张量的最前面添加一个额外维度,以匹配模型的输入形状要求。 torch.unsqueeze(i,dim=0) #按dim=0

卸载cpu版本的torch并离线安装对应的gpu版本

        每次从github上安装项目对应的库,利用requirements.txt安装很容易出现版本不对应的情况,尤其是将torch的gpu版本安装成cpu。这里记录一些查看版本的指令和离线安装的方法,就不用每次百度啦!(注:其他库的离线安装也可以用同样的方法,只需要去相应的网站下载wheel即可)1.首先查看torch版本(指令最好在终端一行一行地输入)activatepytorch#进入环境pythonimporttorchprint(torch.__version__)#查看torch版本print(torch.cuda.is_available())#查看cuda能用吗如果版本

AttributeError: module ‘torch‘ has no attribute ‘concat‘

  如图上所示报错地方在跑算法代码的时候,发现报错,但是这个错误在网上没有找到,我推测是pytorch改版问题,于是查看torch版本改动,发现torch.concat改版后该写为torch.cat。不过或许我写的也不够准确,除此之外还看到了有人问torch.concat和torch.cat的区别。  不过出现了这类问题改成torch.cat试试看吧。

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal

今天在跑yolov7的时候遇见,模型加载问题,因为我是使用CPU来加载pt模型的,但是出现了错误;RuntimeError:AttemptingtodeserializeobjectonaCUDAdevicebuttorch.cuda.is_available()isFalse.IfyouarerunningonaCPU-onlymachine,pleaseusetorch.loadwithmap_location=torch.device('cpu')tomapyourstoragestotheCPU.模型是使用CUDA跑的,但是加载是使用CPU加载的,报错的意思就是需要是需要反序列化加载,

【Pytorch】torch.nn.LeakyReLU()

目录简介torch.nn.LeakyReLU()语法作用举例参考结语简介Hello!非常感谢您阅读海轰的文章,倘若文中有错误的地方,欢迎您指出~ ଘ(੭ˊᵕˋ)੭昵称:海轰标签:程序猿|C++选手|学生简介:因C语言结识编程,随后转入计算机专业,获得过国家奖学金,有幸在竞赛中拿过一些国奖、省奖…已保研学习经验:扎实基础+多做笔记+多敲代码+多思考+学好英语! 唯有努力💪 本文仅记录自己感兴趣的内容torch.nn.LeakyReLU()语法torch.nn.LeakyReLU(negative_slope=0.01,inplace=False)作用构建一个LeakyReLU函数,明确此函数中的

【Pytorch】torch.nn.LeakyReLU()

目录简介torch.nn.LeakyReLU()语法作用举例参考结语简介Hello!非常感谢您阅读海轰的文章,倘若文中有错误的地方,欢迎您指出~ ଘ(੭ˊᵕˋ)੭昵称:海轰标签:程序猿|C++选手|学生简介:因C语言结识编程,随后转入计算机专业,获得过国家奖学金,有幸在竞赛中拿过一些国奖、省奖…已保研学习经验:扎实基础+多做笔记+多敲代码+多思考+学好英语! 唯有努力💪 本文仅记录自己感兴趣的内容torch.nn.LeakyReLU()语法torch.nn.LeakyReLU(negative_slope=0.01,inplace=False)作用构建一个LeakyReLU函数,明确此函数中的

【深度学习】多卡训练__单机多GPU方法详解(torch.nn.DataParallel、torch.distributed)

【深度学习】多卡训练__单机多GPU详解(torch.nn.DataParallel、torch.distributed)文章目录【深度学习】多卡训练__单机多GPU详解(torch.nn.DataParallel、torch.distributed)1.介绍2.单机多GPUの方法2.1方法1:torch.nn.DataParallel2.1.1API2.1.2特点2.1.3例子与解释2.1.4说明2.2方法2:torch.nn.parallel.DistributedDataParallel2.2.1API2.2.2注意事项2.2.3主要代码(可以参照改成自己的)2.2.4对比3.单机多卡训

YOLO UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing a

在运行yolo时出现了一个警告return_VF.meshgrid(tensors,**kwargs) #type:ignore[attr-defined]解决方法:找到pyrcharm所用的虚拟环境下的functional.py文件 具体可以根据报错的提示找到functional的504行加上如下代码 indexing='ij' 保存,问题解决!