码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • Anaconda+PytorchGPU版本+CUDA+CUNN的下载与安装使用


    文章目录

    • 1安装anaconda(略,见我的其他文件)
    • 2安装CUDA和cudnn
      • 2.1确认电脑显卡版本
      • 2.2通过显卡驱动确定驱动版本、cuda核心数以及本机所支持的cuda版本
      • 2.3从官网查找你显卡可用的cuda版本
      • 2.4cuda安装步骤
      • 2.5安装cudnn
    • 3利用anaconda安装pytorch GPU版本

    Anaconda+PytorchGPU版本+CUDA+CUNN的下载与安装使用

    1安装anaconda(略,见我的其他文件)

    2安装CUDA和cudnn

    2.1确认电脑显卡版本

    在安装cuda前,我们要确定本机是否有独立显卡。在计算机->管理->设备管理器(在计算机管理下的系统工具中)->显示适配器。



    如上图,我有两个设备驱动,其中第一个是主板自带的集成显卡,而第二个就是独立显卡,是英伟达getforce GTX 1050

    注意
    如果你的电脑有独立显卡,但是并没有在设备管理中显示出来,此时你应当考虑更新设备驱动试试。
    地址如下:https://www.nvidia.cn/geforce/drivers/

    2.2通过显卡驱动确定驱动版本、cuda核心数以及本机所支持的cuda版本

    打开NVIDIA驱动:点击菜单中的“帮助”->“系统信息”->系统信息->显示/组件

    从这里,可以看出你的显卡的驱动版本、以及显卡所支持的cuda版本以及核心数,这里显示的核心数是640,cuda支持版本是9.2

    2.3从官网查找你显卡可用的cuda版本

    地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

    从这个地址中,找到上面这个表格,可以看到我的显卡可以支持框中这两个版本的GPU。

    地址:https://developer.nvidia.com/cuda-toolkit-archive
    然后这个地址中,找到对应的CUDA,进入后根据系统,选择版本,并进行下载。


    这里还有一个更新文件一并下载后安装即可。

    2.4cuda安装步骤

    1打开安装文件,这个地址不要变,要不然配置环境很麻烦。

    2选择自定义安装,并勾掉vs

    3安装完成后配置环境变量如下图

    4 测试cuda安装是否成功
    配置完成环境变量之后,重启电脑,然后执行下面的命令
    ctrl+r输入cmd调出命令行:输入nvcc -V

    如下图,显示版本则安装成功。

    2.5安装cudnn

    要选择和cuda对应版本的,其实这只是一个基于cuda的库,不需要安装,下载后的压缩包解压后是一些头文件,lib和dll(windows操作系统)文件。
    cudnn下载地址:(需要登录)
    https://developer.nvidia.com/rdp/cudnn-download
    注册登录后,勾选“I agree …”,然后点击对应的版本下载,下载解压后就是一些放到文件夹里的一些包文件。

    然后将这三个文件夹内的文件,分别复制到cuda的安装目录中的名称相同的文件夹内即可。

    3利用anaconda安装pytorch GPU版本

    进入pytorch官网https://pytorch.org/,按照系统选择gpu版本的pytorch,注意要用python3.6环境,否则可能出现兼容性问题。复制conda代码,到conda环境的terminal(终端)中即可安装。

    在terminal中输入:

    conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
    
    • 1
  • 相关阅读:
    ununtu中vim的使用
    STM32WL开发之易智联LORA评估板上定时器TIMER的配置与应用
    layui 使用xmSelect组件多选下拉
    最新版一媒体7.3、星媒体、皮皮剪辑,视频MD ,安卓手机剪辑去重神器+搬运脚本+去视频重软件工具
    【OpenCV-Python】教程:3-8 图像金字塔
    linux安装Sentinal1.8.6
    代码检查过程中为什么需要涉及到编译呢?
    Docker下如何构建包含延迟插件的RabbitMQ镜像
    【从零学习python 】70.网络通信方式及其应用:从直接通信到路由器连接多个网络
    用print学python-列表切片操作,元组定义,遍历
  • 原文地址:https://blog.csdn.net/adreammaker/article/details/125505984
  • 最新文章
  • 攻防演习之三天拿下官网站群
    数据安全治理学习——前期安全规划和安全管理体系建设
    企业安全 | 企业内一次钓鱼演练准备过程
    内网渗透测试 | 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号