windows下使用nvidia GPU机器学习

2024-08-08

windows下使用nvidia GPU机器学习

需要安装:
- CUDA toolkit
- cuDNN

1 查看CUDA版本

1	nvidia-smi

2 下载 CUDA toolkit

下载

网址: https://developer.nvidia.com/cuda-toolkit-archive

选择

3 下载 cuDNN

下载:

网址: https://developer.nvidia.com/rdp/cudnn-archive
选择

4 安装

4.1 安装cuda

双击下载的CUDA toolkit的exe文件, 一直往下走 -> 自定义安装

4.2 安装cudnn

cuDNN已经被集成在Pytorch中，无需单独安装，但是如果需要从源码编译就还是需要安装cuDNN
将下载的cudnn的压缩包里的文件夹里面的文件复制到cuda toolkit对应的同名文件夹下

5 验证cuda是否安装成功

如果终端一直没关过, 可以试试关闭再打开, 刷新一下.

nvcc -V

6 安装pytorch GPU版

缺失模块。
1、请确保node版本大于6.2
2、在博客根目录（注意不是yilia根目录）执行以下命令：
npm i hexo-generator-json-content --save

3、在根目录_config.yml里添加配置：

  jsonContent:
    meta: false
    pages: false
    posts:
      title: true
      date: true
      path: true
      text: false
      raw: false
      content: false
      slug: false
      updated: false
      comments: false
      link: false
      permalink: false
      excerpt: false
      categories: false
      tags: true

很惭愧<br><br>只做了一点微小的工作<br>谢谢大家