我们登上的并|

虾饺爱下棋

园龄:1年6个月粉丝:1关注:2

深度学习中CUDA环境安装教程

首先说明,本人是小白,一次安装,可能有不对的地方,望包含。

安装CUDA

因为我们是深度学习,很多时候要用到gpu进行训练,所以我们需要一种方式加快训练速度。
通俗地说,CUDA是一种协助“CPU任务分发+GPU并行处理”的编程模型/平台,用于加速GPU和CPU之间的计算。
不废话,直接来:

  • 首先检查自己英伟达对应的版本号,可以win+R输入cmd打开终端,输入nvidia-smi
    如下图:

a
可以清楚的看到自己的版本以及Driver Version,同时我们需要借助官方版本对应表进行对应安装。对应链接:安装对照表
如图所示:a

  • 找到对应的版本,即可进行安装。
    安装网址:cuda-toolkit-archive
    注意事项:
    a该图中的version 10和11表示win10和win11的意思。local表示本地下载,network表示网络下载,这个选项选什么都行。
    同时需要注意的是,在安装时会设置两次安装路径,一次是解压,一次是安装,两次路径注意不要放到同一个文件夹中!!!(解压的路径用完后会消失)
    安装时可能会弹窗下图,不用管,继续就行。a

  • 选择自定义安装
    a

  • 可以先试下全选,是否可以安装成功。
    a

  • 若显示安装错误,则可以用以下方法,百试百灵

    • 第一次只安装CUDA,但是不安装其中的Nsight VSE和Visual Studio Integration;其他的不选。

a

  • 然后安装完后,再安装一下CUDA中的Visual Studio Integration即可。
    a
    这一步骤借鉴了博客:NVIDIA cuda安装时全部失败(大佬nb!!!)

  • 最后一步如何检测安装完毕?
    在此win+r输入cmd,从终端中输入nvcc --version
    若显示版本信息,应该安装完成了。a
    本人是小白,一次安装,可能有不对的地方,望包含。

本文作者:虾饺爱下棋

本文链接:https://www.cnblogs.com/lizhongzheng/p/18653793

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   虾饺爱下棋  阅读(109)  评论(1编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起
  1. 1 倔强 五月天
倔强 - 五月天
00:00 / 00:00
An audio error has occurred.