子墨的博客

总得让实力配上野心


  • 首页

  • 标签71

  • 分类16

  • 归档29

  • 关于

  • 搜索

win10下安装cuda+cudnn+pytorch

置顶 发表于 2020-07-21 分类于 机器学习 阅读次数:
本文字数: 1.1k 阅读时长 ≈ 1 分钟

准备

  1. 桌面右键选择NVIDIA控制面板,然后依次选择帮助->系统信息->组件
    查看电脑CUDA驱动适配版本
  2. 可以看到我这台笔记本支持的cuda版本是10.2.95,记住这个
    下载cuda
  3. 然后去nvidia官网下载对应的驱动程序包
    选择对应的cuda版本
  4. 然后去下载对应的cudnn
    下载cudnn,下载cudnn需要NVIDIA账号,如果没有,建议使用163邮箱注册一个(为什么呢?因为QQ邮箱现在收不到NVIDIA的邮件,这就不知道到底是腾讯的问题还是英伟达的问题,亦或者是两者都有问题,手动狗头)
  5. 单独提一嘴,如果下载速度很慢的话,不要奇怪,国外的网站是这样的,想提高下载速度,请挂科学上网,或者是自己找大佬们下载好放在网盘里的资源

安装cuda

  1. 直接双击运行下载好的exe就行
    下载好的cuda和cudnn
  2. 基本都是下一步下一步下一步,没啥好说的
  3. 验证cuda,命令nvcc -V,看到如下图所示就行了
    验证cuda

配置cudnn

  1. 解压cudnn压缩包
    cudnn解压后
  2. 直接 Ctrl + A 全选,然后 Ctrl + C 复制,然后进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目录下(请注意根据版本等选择对应的目录,不要无脑就直接用我这个目录), Ctrl + V 粘贴

安装pytorch

  1. 去pytorch官网,根据操作系统,python包管理等选择对应的选项
    选择pytorch安装命令
  2. 复制上面选择后出现的安装命令,我这里选的是pip,命令行执行pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
  3. 等待安装完成,可能会比较慢,请挂科学上网

验证pytorch

  1. 进入python交互式解释器,cmd中执行python命令即可
    python交互式解释器

  2. 参考官方示例进行验证
    官方示例:

    1
    2
    3
    4
    from __future__ import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)

    输出:

    1
    2
    3
    4
    5
    tensor([[0.3380, 0.3845, 0.3217],
    [0.8337, 0.9050, 0.2650],
    [0.2979, 0.7141, 0.9069],
    [0.1449, 0.1132, 0.1375],
    [0.4675, 0.3947, 0.1426]])
  3. 验证cuda
    官方示例:

    1
    2
    import torch
    torch.cuda.is_available()

    输出:

    1
    True
  4. 验证通过,enjoy it!

相关文章
  • jupyter初体验
觉得不错,打赏一下
子墨 微信支付

微信支付

子墨 支付宝

支付宝

  • 本文作者: 子墨
  • 本文链接: https://blog.zimo.wiki/posts/3908d582/
  • 版权声明: 本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!
机器学习 cuda pytorch 环境搭建
ubantu配置安装tesseract-ocr
记一次排查jetbrains系列ide(如idea,pycharm等) alt+f12 快捷键被占用问题解决过程
  • 文章目录
  • 站点概览
子墨

子墨

子墨的博客
29 日志
16 分类
71 标签
RSS
GitHub E-Mail CSDN QQ Gitee
友情链接
  • 高正杰的博客

Tag Cloud

  • 8076
  • HttpCanary1
  • JavaScript2
  • Jupyter Notebook1
  • c++1
  • centos1
  • cuda1
  • c语言6
  • deepin1
  • dns2tcp1
  • fiddler1
  • hexo2
  • html1
  • i至诚1
  • jar1
  • java3
  • jetbrains1
  • linux3
  • linux server1
  • markdown1
  • nginx1
  • nodejs1
  • python2
  • python31
  • pytorch1
  • tesseract-ocr1
  • ubantu1
  • virtualenvwrapper-win1
  • war1
  • windows2
  • windows server1
  • 个人博客2
  • 代理1
  • 代码托管1
  • 代码雨1
  • 伪装位置1
  • 使用指南1
  • 刷recovery1
  • 力扣1
  • 劫持1
  • 双系统1
  • 小爱课程表1
  • 小米61
  • 常识1
  • 快捷键冲突1
  • 抓包1
  • 折腾1
  • 挖矿木马1
  • 服务器1
  • 机器学习2
  • 极客1
  • 树梅派4001
  • 油猴脚本1
  • 爬虫1
  • 环境搭建1
  • 直播服务器1
  • 科普1
  • 程序综合设计6
  • 算法1
  • 终端1
  • 编译1
  • 考研7
  • 自动打卡1
  • 蓝桥杯1
  • 解锁bl1
  • 运维1
  • 部署1
  • 钉子户1
  • 题解1
  • 黑客帝国1
  • 黑苹果1
  1. 1. 准备
  2. 2. 安装cuda
  3. 3. 配置cudnn
  4. 4. 安装pytorch
  5. 5. 验证pytorch
蜀ICP备18029083号 © 2019 – 2022 子墨 | 站点总字数: 167k | 站点阅读时长 ≈ 2:32
由 Hexo 强力驱动 v3.9.0
|
主题 – NexT.Pisces v7.3.0
载入网站运行时间中...
|
0%