草庐IT

Pytorch-CUDA

全部标签

Pytorch中GPU相关操作

一、GPU基本信息1.查看cuda是否可用:torch.cuda.is_available()>>>importtorch>>>torch.cuda.is_available()True2.查看gpu数量:torch.cuda.device_count()>>>torch.cuda.device_count()33.查看gpu名字,设备索引默认从0开始:torch.cuda.get_device_name(0)>>>torch.cuda.get_device_name(0)'TeslaP40'4.当前设备索引:torch.cuda.current_device()>>>torch.cuda.c

强化学习问题(7)--- Python和Pytorch,Tensorflow的版本对应

1.问题   之前下载的python3.8,在对应Pytorch和Tensorflow时没太在意版本,在运行一些代码时,提示Pytorch和Tensorflow版本过高,直接降下来,有时候又和Python3.8不兼容,所以又在虚拟环境搞一个Pyhon3.7,下载一些低版本的Pytorch和Tensorflow。   代码环境要求如下:2.解决  2.1Pytorch版本对应1.Python安装torch(含torch和torchvision对应版本)_你好,明天,,的博客-CSDN博客2.1Tensorflow版本对应 安装代码pipinstalltensorflow==1.14.0-ihtt

text-generation-webui加载codellama报错DLL load failed while importing flash_attn_2_cuda: 找不到指定的模块。

使用text-generation-webui加载codellama,报错:Traceback(mostrecentcalllast):File"C:\Users\Ma\AppData\Roaming\Python\Python310\site-packages\transformers\utils\import_utils.py",line1353,in_get_modulereturnimportlib.import_module("."+module_name,self.__name__)File"D:\Anaconda\Anaconda\envs\codellama\lib\impor

卷积神经网络(CNN):基于PyTorch的遥感影像、无人机影像的地物分类、目标检测、语义分割和点云分类

我国高分辨率对地观测系统重大专项已全面启动,高空间、高光谱、高时间分辨率和宽地面覆盖于一体的全球天空地一体化立体对地观测网逐步形成,将成为保障国家安全的基础性和战略性资源。随着小卫星星座的普及,对地观测已具备多次以上的全球覆盖能力,遥感影像也不断被更深入的应用于矿产勘探、精准农业、城市规划、林业测量、军事目标识别和灾害评估。未来10年全球每天获取的观测数据将超过10PB,遥感大数据时代已然来临。另一方面,随着无人机自动化能力的逐步升级,它被广泛的应用于多种领域,如航拍、农业、植保、灾难评估、救援、测绘、电力巡检等。但同时由于无人机飞行高度低、获取目标类型多、以及环境复杂等因素使得对无人机获取的

pytorch一行实现:计算同一tensor矩阵内每行之间的余弦相似度

文章目录0输入数据1余弦相似度(CosineSimilarity)2torch.cosine_similarity3问题4分析与解决4.1答案5另外的实现方法0输入数据importtorch#设置随机数种子,以保证结果可重现torch.manual_seed(0)a=torch.randn(4,3)tensor([[1.5410,-0.2934,-2.1788],[0.5684,-1.0845,-1.3986],[0.4033,0.8380,-0.7193],[-0.4033,-0.5966,0.1820]])1余弦相似度(CosineSimilarity)  余弦相似度的公式如下所示:2to

【Linux】安装CUDA 11.2 和 cuDNN 8.4.0并检查是否安装成功

目录一、前言二、安装CUDA三、安装cuDNN四、总结五、参考一、前言正如题目所言,最近笔者需要配置一台机器上的深度学习环境,即CUDA工具包和与之对应的cuDNN库,具体步骤如下。因为我已经装过了,为了教程演示,首先卸载CUDA工具包,利用其自带的卸载程序:检查本机是否有CUDA工具包,输入nvcc-V:二、安装CUDA在命令行输入nvidia-smi查看显卡驱动版本也就是最高支持的CUDA工具包版本。例如,本机可安装11.2及以下的CUDA工具包:在nVidia官网选择对应版本的CUDA工具包并选择你的机器配置,我们就选择11.2.0版本下载,在终端执行如下命令:wgethttps://d

基于pytorch 的RNN实现文本分类

首先,需要导入必要的库,包括torch、torchtext、numpy等:importtorchimporttorch.nnasnnimporttorch.optimasoptimimportnumpyasnpfromtorchtext.datasetsimportAG_NEWSfromtorchtext.data.utilsimportget_tokenizerfromcollectionsimportCounter然后,我们需要加载数据集并进行数据预处理。在这里,我们使用AGNews数据集,其中包含120,000个新闻文本,分为四个不同的类别:World、Sports、Business和S

【图像分类】【深度学习】【Pytorch版本】 ResNeXt模型算法详解

【图像分类】【深度学习】【Pytorch版本】ResNeXt模型算法详解文章目录【图像分类】【深度学习】【Pytorch版本】ResNeXt模型算法详解前言ResNeXt讲解分组卷积(GroupConverlution)分割-变换-合并策略(split-transform-merge)ResNeXt模型结构ResNeXtPytorch代码完整代码总结前言ResNeXt是加利福尼亚大学圣迭戈分校的Xie,Saining等人在《AggregatedResidualTransformationsforDeepNeuralNetworks【CVPR-2017】》【论文地址】一文中提出的模型,结合Res

pytorch中.numpy()、.item()、.cpu()、.detach()及.data的使用

目录.numpy().item().cpu().detach()和.data(重点).numpy()Tensor.numpy()将Tensor转化为ndarray,这里的Tensor可以是标量或者向量(与item()不同)转换前后的dtype不会改变a=torch.tensor([[1.,2.]])a_numpy=a.numpy()#[[1.,2.]].item()将一个Tensor变量转换为python标量(intfloat等)常用于用于深度学习训练时,将loss值转换为标量并加,以及进行分类任务,计算准确值值时需要optimizer.zero_grad()outputs=model(dat

人工智能(pytorch)搭建模型13-pytorch搭建RBM(受限玻尔兹曼机)模型,调通模型的训练与测试

大家好,我是微学AI,今天给大家介绍一下人工智能(pytorch)搭建模型13-pytorch搭建RBM(受限玻尔兹曼机)模型,调通模型的训练与测试。RBM(受限玻尔兹曼机)可以在没有人工标注的情况下对数据进行学习。其原理类似于我们人类学习的过程,即通过观察、感知和记忆不同事物的特点,从而形成对这些事物的认知模型。本文将介绍RBM(受限玻尔兹曼机)模型的原理,并使用PyTorch框架实现一个简单的RBM模型。我们将介绍如何构建模型,加载样例数据进行训练,以及在训练完成后进行测试。文章目录结构:RBM模型简介RBM模型原理使用PyTorch搭建RBM模型数据样例及加载模型训练模型测试总结1.RB