1. 查看自己电脑显卡驱动对应的cuda版本
    进入NVIDIA控制面板==>帮助==>系统信息==>组件,找到下图,即显卡对应cuda版本
    在这里插入图片描述

  2. 下载cuda程序,下为官网地址,找到对应的CUDA应用程序

    https://developer.nvidia.com/cuda-toolkit-archive
    
  3. 安装CUDA
    ① 默认 ok
    在这里插入图片描述
    ② 默认 同意并继续
    在这里插入图片描述
    ③ 选择自定义
    在这里插入图片描述
    ④选择CUDA,其他的都取消,CUDA中把Visual Studio Integration取消I(如果电脑中安装有可以取消勾选),点击 下一步等待安装完成;
    在这里插入图片描述
    ⑤安装位置,一般选择默认C盘
    在这里插入图片描述

    如果需要修改,需要创建两个文件夹如下图:
    在这里插入图片描述

    NVIDIA Corporation
    NVIDIA GPU Computing Toolkit
    

    ⑥点击下一步,等待安装完成

  4. 安装cudnn,cudnn官网,需要邮箱登陆注册:

    https://developer.nvidia.com/rdp/cudnn-download
    

    找到对应的cudnn,cudnn下载下来是一个压缩包,类似于CUDA的补丁,下载后解压:
    在这里插入图片描述
    上面文件全选,复制到以下文件夹下(CUDA安装目录)

    E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3
    

    在这里插入图片描述

  5. 加入系统变量
    把以下文件夹路径加入到系统变量Path中

    E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
    E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include
    E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp
    E:\cudainstall\NVIDIA Corporation\CUDA Samples\v11.3\common\lib\x64
    E:\cudainstall\NVIDIA Corporation\CUDA Samples\v11.3\bin\win64
    
  6. 测试是否安装成功,打开cmd,输入nvcc -V
    在这里插入图片描述
    安装成功

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐