pytorch-gpu 极简安装

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

1、进入pytoch官网:PyTorch

找到pytorch-gpu版本,看到CUDA11.8、11.7、CPU,这里我选择安装CUDA11.8

pytorch-gpu 极简安装,pytorch,人工智能,python

2、下载CUDA Toolkit:CUDA Toolkit 11.8 Downloads | NVIDIA Developer

3、下载CUDANN:cuDNN Download | NVIDIA Developer

pytorch-gpu 极简安装,pytorch,人工智能,python

命令行输入nvidia-smi,查看驱动信息

输入nvcc --version 进行检查

pytorch-gpu 极简安装,pytorch,人工智能,python

4、根据pytoch官方指令下载pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

5、验证:文章来源地址https://www.toymoban.com/news/detail-621362.html

>>> import torch
>>> torch.__version__
'2.0.0+cu118'
>>> torch.cuda.is_available()
True
>>> torch.cuda.device(0)
<torch.cuda.device object at 0x000001F2F10AAC70>
>>> torch.cuda.device_count()
1
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce GTX 1060 with Max-Q Design'
@著作:机智的小神仙儿

@时间:2023.7.31

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

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

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

相关文章

  • 读十堂极简人工智能课笔记08_人工智能的未来

    1.2.2.1. 其骨头是用塑料生产的,结构相当精巧,足以匹配人类的骨骼 2.2.2.1. 其灵感来自人脑 2.2.2.2. 如今被用于神经科学、机器人和计算机科学 2.2.3.1. 每个处理器又由18个较小的处理器组成 2.2.3.1.1. 16个用于模拟神经元 2.2.3.1.2. 1个用于管理 2.2.3.1.3. 1个备用 2.2.3.2. 其巧妙

    2024年02月21日
    浏览(31)
  • 【人工智能概论】 PyTorch可视化工具Tensorboard安装与简单使用

    Tensorboard原本是Tensorflow的可视化工具,但自PyTorch1.2.0版本开始,PyTorch正式内置Tensorboard的支持,尽管如此仍需手动安装Tensorboard。否则会报错。 ModuleNotFoundError: No module named ‘tensorboard’ 进入相应虚拟环境后,输入以下指令即可安装。 输入以下指令,不报错即说明安装成功。

    2023年04月24日
    浏览(37)
  • 读十堂极简人工智能课笔记07_模拟与情感

    2.5.1.1. 冯·诺伊曼 2.5.1.2. 可以执行计算机可以完成的所有任务 2.5.2.1. 冯·诺伊曼 4.5.2.1. 一位计算机图形专家 4.5.2.2. 1986年创造了一种基于代理的模型算法 4.5.4.1. 最早的例子之一便是电影《蝙蝠侠归来》(1992)中计算机生成的蝙蝠群和企鹅群 4.5.6.1. 克雷格·雷诺兹(2001) 4.7

    2024年02月20日
    浏览(42)
  • 读十堂极简人工智能课笔记06_自然语言处理

    1.4.3.1. 能让真人腾出手来处理难度更大的咨询 2.1.3.1. 在研究儿童的语言能力发展后总结出来的理论 2.1.3.2. 儿童虽然能够学会流利地说话,但他们在学习过程里其实根本没有接收到足够的信息 2.1.3.2.1. 所谓的“刺激的贫乏” 2.1.3.3. 儿童能够发展语言技能的唯一途径是他们

    2024年02月19日
    浏览(39)
  • 读十堂极简人工智能课笔记03_遗传算法与进化

    1.1.3.1. 创造一批能游泳、走路、跳跃,甚至互相竞争的虚拟动物震惊了整个科学界 1.1.3.2. 它们的人工大脑却是个极其复杂的网络,信息经由传感器的输入,经过大量的数学函数计算和操作,才能产生那些看起来很聪明的动作和表现 1.1.4.1. 他并没有设计这些动物 1.1.4.2. 他并

    2024年02月19日
    浏览(34)
  • 读十堂极简人工智能课笔记04_计算机视觉

    3.2.3.1. 应该发现真正的边缘,而尽量避免错报 3.2.4.1. 应该正确地找出边缘的确切位置 3.2.5.1. 每条实际的边缘应该检测为一条边缘,而不是多条边缘 4.7.5.1. 有数以百万计的几乎任何种类的图像例子 4.7.7.1. 神经网络自己就能完成这一切

    2024年02月19日
    浏览(35)
  • 读十堂极简人工智能课笔记02_选对路径与犯错

    1.5.1.1. 不可能把数学里的所有命题都证明出来 1.7.2.1. 这种信念的根基是如下这种思想:人类智能完全就是对符号的操纵 1.8.6.1. 从始至终你都并不理解纸条上说了什么,因为你完全不懂中文 2.1.1.1. 纽厄尔还创造了列表处理的概念,后来成为人工智能里一个重要的语言,叫作

    2024年02月19日
    浏览(38)
  • win10下wsl2使用记录(系统迁移到D盘、配置国内源、安装conda环境、配置pip源、安装pytorch-gpu环境、安装paddle-gpu环境)

    安装好后环境测试效果如下,支持命令nvidia-smi,不支持命令nvcc,usr/local目录下没有cuda文件夹。 系统迁移到非C盘 wsl安装的系统默认在c盘,为节省c盘空间进行迁移。 1、输出 wsl -l 查看要迁移的系统名称 2、执行导出命令: wsl --export Ubuntu-20.04 ./Ubuntu-20.04.tar ,以下命令将系统

    2024年02月20日
    浏览(45)
  • 4070配置pytorch-GPU过程记录

    为新买的电脑配置深度学习环境,记录几个踩坑点 我的电脑信息:4070显卡,最高支持CUDA12.0,已安装pycharm,anaconda并且已经创建环境python=3.10 在安装CUDA之前需要先确定pytorch支持的CUDA,截止到2023.3.3号pytorch最高版本先行版支持11.8(稳定版是11.7,但搜索资料得知 pytorch是从11.

    2024年02月09日
    浏览(39)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包