码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • Pytorch环境配置——从零开始(不需要anaconda)


    Pytorch环境配置——从零开始(不需要anaconda)

    • Pytorch环境配置
      • 查看显卡信息
      • 安装CUDA Toolkit
      • 下载pytorch的whl文件,pip本地安装
      • 测试GPU是否运行
        • pip在线安装

    前面的文章,配置基本环境
    Python 环境安装系统教程——PyCharm
    pytorch学术界用的多,tensorflow工业界用的多。

    网上很多pytorch的教程,都需要安装anaconda,其实不需要花那些时间去装一个anaconda,还占电脑很多空间。

    下面讲一下不用anaconda的环境配置方法。

    Pytorch环境配置

    查看显卡信息

    打开cmd,输入

    nvidia-smi
    
    • 1

    如下图,得到电脑显卡信息,和CUDA版本,
    在这里插入图片描述
    NVDIA显卡驱动, CUDA, cuDNN概念及安装顺序
    显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn区别?

    NVIDIA的显卡驱动程序 和 CUDA完全是两个不同的概念!
    CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。CUDA的本质是一个工具包(ToolKit)。
    当我们使用一台电脑的时候默认的已经安装了NVIDIA的显卡驱动,因为没有显卡驱动根本用不了显卡嘛,但是这个时候我们是没有CUDA可以用的,我们可以更新我们的驱动。
    所以,NVIDIA显卡驱动和CUDA工具包本身是不具有捆绑关系的,也不是一一对应的关系,只不过是离线安装的CUDA工具包会默认携带与之匹配的最新的驱动程序。

    CUDA:为“GPU通用计算”构建的运算平台。
    cudnn:为深度学习计算设计的软件库。
    CUDA Toolkit (nvidia): CUDA完整的工具安装包,其中提供了 Nvidia 驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。包括 CUDA 程序的编译器、IDE、调试器等,CUDA 程序所对应的各式库文件以及它们的头文件。
    CUDA Toolkit (Pytorch): CUDA不完整的工具安装包,其主要包含在使用 CUDA 相关的功能时所依赖的动态链接库。不会安装驱动程序。
    (NVCC 是CUDA的编译器,只是 CUDA Toolkit 中的一部分)

    安装CUDA Toolkit

    1. 配置电脑的显卡:(1)允许独显直连。(2)配置使用GPU
    2. 安装toolkit:在下面的官方网站下载:
      https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local
      在这里插入图片描述
      因为我是11.6/7版本的cuda,所以这里选择11,然后选择local,下载即可。下载完之后直接安装就好了,一定一定不要更改任何选项,不要更改路径!!!
      安装好进入下一步。

    下载pytorch的whl文件,pip本地安装

    pytorch官网

    在pypi搜索torch,下载到的只有cpu版本的,所以另求他法:
    在这篇文章中附了这个链接下载,https://download.pytorch.org/whl/torch_stable.html这才是对的,
    pytorch-whl下载地址和安装过程(附下载链接)
    我是cuda11.7,兼容11.6,所以下载11.6的即可。
    在这里插入图片描述
    下载完之后,将这个利用pip安装即可。安装完,pytorch的环境就算配置成功了。

    测试GPU是否运行

    进入cmd之后

    python
    
    • 1

    进入python环境
    然后导入torch

     import torch
    
    • 1

    然后输入测试语句

    torch.cuda.is_available()
    
    • 1

    输出为true则正确。
    在这里插入图片描述
    安装成功了。

    pip在线安装

    这个需要去官网,选择要安装的版本等等,然后网站上会生成一条pip语句,复制到命令框就可以下载了。

  • 相关阅读:
    SSL VPN综合实验
    JavaScript数据结构与算法
    巧用大数据分析工具,让本地数据输入速度提升2倍的方法
    阿里巴巴面试题- - -Java体系最新面试题(5)
    适合短期内想快速上手数字孪生系统的人使用的数字孪生软件推荐
    AVL树到底是什么?
    Linux/Ubuntu环境搭建(二):创建添加新磁盘、搭建Samba服务器
    Spark repartitionAndSortWithinPartitions
    手把手教你语音识别(二)
    [附源码]Python计算机毕业设计Django的低碳生活记录网站
  • 原文地址:https://blog.csdn.net/lgyLGY35/article/details/126796106
  • 最新文章
  • 攻防演习之三天拿下官网站群
    数据安全治理学习——前期安全规划和安全管理体系建设
    企业安全 | 企业内一次钓鱼演练准备过程
    内网渗透测试 | Kerberos协议及其部分攻击手法
    0day的产生 | 不懂代码的"代码审计"
    安装scrcpy-client模块av模块异常,环境问题解决方案
    leetcode hot100【LeetCode 279. 完全平方数】java实现
    OpenWrt下安装Mosquitto
    AnatoMask论文汇总
    【AI日记】24.11.01 LangChain、openai api和github copilot
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1
正则表达式工具 cron表达式工具 密码生成工具

京公网安备 11010502049817号