草庐IT

pytorch_model

全部标签

【超详细小白必懂】Pytorch 直接加载ResNet50模型和参数实现迁移学习

Torchvision.models包里面包含了常见的各种基础模型架构,主要包括以下几种:(我们以ResNet50模型作为此次演示的例子)AlexNetVGGResNetSqueezeNetDenseNetInceptionv3GoogLeNetShuffleNetv2MobileNetv2ResNeXtWideResNetMNASNet首先加载ResNet50模型,如果如果需要加载模型本身的参数,需要使用pretrained=True,代码如下importtorchvisionfromtorchvisionimportmodelsresnet50=models.resnet50(pretra

GPU版pytorch安装成功却无法使用cuda

在远程服务器安装pytorch,根据官网命令进行安装,但在完成之后,显示GPU不可用,故记录此大坑。一、根据官网进行安装 安装的很快,但是!!安装结束之后,输入以下代码进行安装验证却显示没有成功安装!!importtorch#如果pytorch安装成功即可导入print(torch.cuda.is_available())#查看CUDA是否可用print(torch.cuda.device_count())#查看可用的CUDA数量print(torch.version.cuda)#查看CUDA的版本号所以这表明安装大失败!但是不死心的我又输入以下语句来检查torchcondalist这表明我们

【Stable Diffusion论文精读】High-Resolution Image Synthesis with Latent Diffusion Models(主打详细和易懂)

【StableDiffusion论文精读】High-ResolutionImageSynthesiswithLatentDiffusionModels(主打详细和易懂)0、前言(学的明明白白)Abstract1.Introduction1.1民主化的Democratizing高分辨率图像合成1.2向潜在空间出发1.3总结2.RelatedWork(粗看)2.1GenerativeModelsforImageSynthesis2.2DiffusionProbabilisticModels(DM)2.3Two-StageImageSynthesis3.Method(需要细看)3.1.Percept

深度学习环境搭建超级详解(Miniconda、pytorch安装)

小白刚开始学习《动手学深度学习》,第一次发文,本文主要是为了记录在环境搭建过程中遇到的问题和疑惑,以及解决方法,同时希望能帮到遇到相同问题的小伙伴。在学习中遇到的疑惑和最后搜索得到的解答,都放在了每一段的Q&A中了,若遇到问题可查看。另外特别感谢在搭建过程中从此答主处收获颇丰,本文许多内容与环境搭建实现过程都是对此教程的复现:阖家欢乐:搭建深度学习平台的详细过程(pytorch安装)https://zhuanlan.zhihu.com/p/393409713话不多说,开干:本次环境搭建所需安装的有:Miniconda、CUDA、cuDNN、torch、torchvision、pycharm首要

深度学习环境搭建超级详解(Miniconda、pytorch安装)

小白刚开始学习《动手学深度学习》,第一次发文,本文主要是为了记录在环境搭建过程中遇到的问题和疑惑,以及解决方法,同时希望能帮到遇到相同问题的小伙伴。在学习中遇到的疑惑和最后搜索得到的解答,都放在了每一段的Q&A中了,若遇到问题可查看。另外特别感谢在搭建过程中从此答主处收获颇丰,本文许多内容与环境搭建实现过程都是对此教程的复现:阖家欢乐:搭建深度学习平台的详细过程(pytorch安装)https://zhuanlan.zhihu.com/p/393409713话不多说,开干:本次环境搭建所需安装的有:Miniconda、CUDA、cuDNN、torch、torchvision、pycharm首要

Faster-RCNN模型跑通总结(使用pytorch1.10+cuda10.2版本)

Faster-RCNN模型搭建跑通总结0、前言1、准备操作系统2、安装驱动及cuda2.1、安装驱动2.2、安装cuda3、安装anaconda和pytorch3.1安装anaconda3.1.1为什么推荐安装anaconda而不是pip安装?3.1.2安装anaconda3.1.3配置国内镜像源3.2安装pytorch3.2.1确认要安装的pytorch版本3.2.2安装pytorch3.2.2.1在conda中创建虚拟环境3.2.2.2激活该虚拟环境3.2.2.3在该环境中安装pytorch3.2.2.4确认安装的pytorch版本4、训练faster-rcnn模型4.1下载模型到本地4.

mmcv与cuda,pytorch版本匹配要求

mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip安装部分。目前网页上默认最新版2.x版本,若要切换旧版,点击页面左下角切换即可。查看自己的cuda和torch版本:python-c'importtorch;print(torch.__version__);print(torch.version.cuda)'#pytorch2.0版本需要cuda11.7及以上点击文档链接选择自己所需版本,拷贝对应的安装命令,进行安装注意!!!op

Anaconda安装pytorch遇到一直循环Solving environment问题+更换镜像源

原因:访问的国外的url,很慢要更换一下镜像源解决方式:找到这个目录下的这个文件,图里面的“1234”是我的用户名右键打开,打开方式可以用记事本,把下面的复制进去channels:-http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/-http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/-http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/show_channel_urls:truessl_verify:true完

苹果mac m1,m2芯片安装 pytorch和tensorflow的GPU版本

一、下载M芯片的anaconda,并安装二、安装GPU版本的pytorch1.安装Xcodexcode-select--install2.创建环境condacreate-ntorch-gpupython=3.9condaactivatetorch-gpu3.打开pytorch官网复制命令, 注意:在macm上,device是’mps’而不是’cuda’,mac的MPS支持MacOS12.3+ 4.测试importtorchimportmathprint(torch.backends.mps.is_available())#Trueprint(torch.backends.mps.is_buil

Dr. LLaMA: Improving Small Language Models in Domain-Specific QAvia Generative Data Augmentation

https://arxiv.org/pdf/2305.07804.pdfhttps://arxiv.org/pdf/2305.07804.pdfOurfindingsindicatethatLLMseffectivelyrefineanddiversifyexistingquestion-answerpairs,resultinginimprovedperformanceofamuchsmallermodelondomain-specificQAdatasetsafterfine-tuning.ThisstudyhighlightsthechallengesofusingLLMsfordoma