关于安装 PyTorch-Lightning 的一些问题(GPU版)

这篇具有很好参考价值的文章主要介绍了关于安装 PyTorch-Lightning 的一些问题(GPU版)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

官网地址:

PyTorch

PyTorch-Lightning

安装PyTorch-Lightning

1、不能直接使用pip install pytorch-lightning ,否则如下图会直接卸载掉你的torch而安装cpu版本的torch。

Installing collected packages: torch, lightning_fabric
  Attempting uninstall: torch
    Found existing installation: torch 1.9.1+cu111
    Uninstalling torch-1.9.1+cu111:
      Successfully uninstalled torch-1.9.1+cu111

2、在安装pytorch-lightning时一定注意自己的torch是pip安装还是conda安装,两者要保持一致,否则也会导致你的torch版本被替换。

正确安装方式:pip install pytorch-lightning==版本名

验证PyTorch和PyTorch-Lightning版本

 在Python中导入PyTorch,然后打印出其版本号来查看,示例代码如下:

import torch
print(torch.__version__)

使用以下代码来查看已安装的 PyTorch-Lightning 版本: 

import pytorch_lightning as pl
print(pl.__version__)

验证GPU是否能用

import torch

if torch.cuda.is_available():
    print("CUDA is available")
else:
    print("CUDA is not available")
import pytorch_lightning as pl
print(pl._C.has_cuda) # will print True if GPU is supported, False otherwise

 一些报错解决:

import pytorch_lightning出现的报错:

 ImportError: cannot import name ‘get_num_classes’ from ‘torchmetrics.utilities.data’

根据参考  ImportError: cannot import name ‘get_num_classes’ from ‘torchmetrics.utilities.data’在虚拟环境下执行下面命令

pip uninstall torchmetrics
pip install torchmetrics==0.5

importerror: cannot import name 'get_num_classes' from 'torchmetrics.utiliti,python,pip,conda,pytorch,深度学习

 然而,出现了新的报错

AttributeError: module 'distutils' has no attribute 'version'

根据参考python, torch. 遇到AttributeError: module ‘distutils‘ has no attribute ‘version‘ 报错。和AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案

在虚拟环境下执行下面命令

pip install setuptools==59.5.0

importerror: cannot import name 'get_num_classes' from 'torchmetrics.utiliti,python,pip,conda,pytorch,深度学习

再次尝试,报错解决

importerror: cannot import name 'get_num_classes' from 'torchmetrics.utiliti,python,pip,conda,pytorch,深度学习

参考文章:

pytorch-lightning安装

安装 pytorch-lightning 的坑

 ImportError: cannot import name ‘get_num_classes’ from ‘torchmetrics.utilities.data’

 python, torch. 遇到AttributeError: module ‘distutils‘ has no attribute ‘version‘ 报错。

 AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案

AttributeError: module 'distutils' has no attribute 'version'文章来源地址https://www.toymoban.com/news/detail-783250.html

到了这里,关于关于安装 PyTorch-Lightning 的一些问题(GPU版)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • pytorch lightning 入门

    翻译自官方文档 前置知识 :推荐pytorch 目标 :通过PL中7个关键步骤了解PL工作流程 PL是基于pytorch的高层API,自带丰富的工具为AI学者和工程师快速创建高性能模型,去除繁琐的重复流程同时保持灵活性。 使用组织好的pytorch代码,PL可以: 避免重复流程。比如gpu设置,device设

    2023年04月08日
    浏览(67)
  • Pytorch Lightning 训练更新次数

    假设一共1000个samples,batch size=4,因此一个epoch会有250 iterations,也就是会更新250次 当设置Trainer时 这个 max_steps 指的是最多更新的次数,这里也就是40次,而 accumulate_grad_batches 指的是每次更新前积累多少个batch,这里为2 因此,每次更新前实际上积累了2 * 4 = 8个samples的gradient

    2024年02月15日
    浏览(45)
  • PyTorch Lightning教程五:Debug调试

    如果遇到了这样一个问题,当一次训练模型花了好几天,结果突然在验证或测试的时候崩掉了,这个时候其实是很奔溃的,主要还是由于没有提前知道哪些时候会出现什么问题,本节会引入Lightning的Debug方案 1.fast_dev_run参数 Trainer中的fast_dev_run参数通过你的训练器运行5批训练

    2024年02月14日
    浏览(44)
  • PyTorch Lightning教程七:可视化

    本节指导如何利用Lightning进行可视化和监控模型 为何需要跟踪参数 在模型开发中,我们跟踪感兴趣的值,例如validation_loss,以可视化模型的学习过程。模型开发就像驾驶一辆没有窗户的汽车,图表和日志提供了窗口,让我们知道该把车开到哪里。有了Lightning,几乎可以可视

    2024年02月14日
    浏览(38)
  • PyTorch Lightning教程四:超参数的使用

    如果需要和命令行接口进行交互,可以使用Python中的argparse包,快捷方便,对于Lightning而言,可以利用它,在命令行窗口中,直接配置超参数等操作,但也可以使用LightningCLI的方法,更加轻便简单。 ArgumentParser ArgumentParser是Python的内置特性,进而构建CLI程序,我们可以使用它

    2024年02月15日
    浏览(34)
  • PyTorch Lightning教程八:用模型预测,部署

    关于Checkpoints的内容在教程2里已经有了详细的说明,在本节,需要用它来利用模型进行预测 加载checkpoint并预测 使用模型进行预测的最简单方法是使用LightningModule中的load_from_checkpoint加载权重。 predict_step方法 加载检查点并进行预测仍然会在预测阶段的epoch留下许多boilerplate,

    2024年02月12日
    浏览(39)
  • 关于信号的一些问题

    信号在操作系统中是非常重要的概念之一,因此在面试中经常会涉及到与信号相关的问题。以下是一些可能被问到的面试题: 什么是信号? 答:信号是进程间通信的一种方式,它是一种软件中断机制,用于通知进程发生了某个事件。信号可以由内核、进程本身或其他进程发

    2024年02月07日
    浏览(41)
  • PyTorch Lightning教程二:验证、测试、checkpoint、早停策略

    介绍:上一期介绍了如何利用PyTorch Lightning搭建并训练一个模型(仅使用训练集),为了保证模型可以泛化到未见过的数据上,数据集通常被分为训练和测试两个集合,测试集与训练集相互独立,用以测试模型的泛化能力。本期通过增加验证和测试集来达到该目的,同时,还

    2024年02月16日
    浏览(34)
  • 变分自编码器(VAE)PyTorch Lightning 实现

    ✅作者简介:人工智能专业本科在读,喜欢计算机与编程,写博客记录自己的学习历程。 🍎个人主页:小嗷犬的个人主页 🍊个人网站:小嗷犬的技术小站 🥭个人信条:为天地立心,为生民立命,为往圣继绝学,为万世开太平。 变分自编码器 (Variational Autoencoder,VAE)是一

    2024年02月21日
    浏览(51)
  • (5)深度学习学习笔记-多层感知机-pytorch lightning版

    pytorch lighting是导师推荐给我学习的一个轻量级的PyTorch库,代码干净简洁,使用pl更容易理解ML代码,对于初学者的我还是相对友好的。 pytorch lightning官网网址 https://lightning.ai/docs/pytorch/stable/levels/core_skills.html 代码如下: 代码如下:(可以直接把download改为true下载) 更多pl的方

    2024年02月12日
    浏览(44)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包