GPU - cuda 安装

GPU - cuda 安装

环境搭建安装

0.确认你的电脑上有英伟达显卡

  • 通过win+R输入:control /name Microsoft.DeviceManager
  • 打开显示适配器,能看到显卡即可。
  • 我的版本是 3060 驱动版本 31.0.15.4617

1.查看主机显卡驱动版本. 主机GPU驱动版本决定你的主机最高能支持到什么版本的cuda程序

  • NVIDIA控制面板中点击,帮助 --- 系统信息 --- 驱动程序版本
  • 经查看, 我主机3060 12G显卡,版本是 546.12
    GPU - cuda 安装_第1张图片
    GPU - cuda 安装_第2张图片

2.根据主机显卡驱动版本来确定主机应该安装哪个版本的CUDA Tookit

  • 打开CUDA Tookit和GPU驱动版本对照表
  • 找到表格3:Table 3 CUDA Toolkit and Corresponding Driver Versions
  • 根据第一条,我得驱动版本为546.12,所以我的电脑最高支持的版本为: CUDA 12.2 GA
    GPU - cuda 安装_第3张图片

3.下载 CUDA Tookit

  • 默认下载最新版本(截止文章发布为V12.3)
  • 可以点开查看Archive of Previous CUDA Releases ,选一个版本下载
  • 双击安装即可

4.设置环境变量

  • CUDA_PATH : C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
  • %CUDA_PATH% 以及 %CUDA_PATH%\bin
  • 输入指令nvcc --version 检测是否成功
    GPU - cuda 安装_第4张图片

下载 CUDA Tookit 方式二

有两种方法下载安装。一种是通过上文提到的手动下载并安装。还有一种是通过conda命令行下载并安装。

在这里我使用conda命令行进行演示:

1.下载安装 Anaconda

2.安装指定版本cuda(11.8)

  • 执行命令: conda install cuda -c nvidia/label/cuda-11.8.0
  • cuda-11.8这个版本的来源

3.检查cuda是否安装完毕

  • 输入指令: nvcc -V

检测cuda是否安装成功

import torch
print(torch.cuda.is_available())
print(torch.__version__)  #注意是双下划线

pytorch版本,cuda版本,系统cuda版本查询和对应关系

1.pytorch和cudatoolkit版本对应关系

2.cudatoolkit版本和系统cuda对应关系:

3.系统cuda和nvidia对应关系:

4.cuda和cuDNN的关系和对应关系:

  • https://www.jianshu.com/p/622f47f94784

tips

  • 一台主机上,要使用GPU,那么至少要有一个GPU驱动程序
  • 一台主机上,可以安装多个cuda 工具(例如我主机安装了11.8以及12.3)
  • 查看主机最高支持的gpu版本: 命令行输入nvidia-smi,找到CUDA Version
  • 卸载 CUDA 软件。可以通过使用 Windows 控制面板使用“程序和功能”小部件卸载

参考

  • CUDA版本和显卡驱动版本对照表

  • 显卡驱动下载

  • pytorch官方

  • 官网-安装cuda教程

  • 参考安装文章1

  • 参考安装文章2

  • 参考安装文章3

  • 参考安装文章4

  • 参考安装文章5 -环境变量


------ 如果文章对你有用,感谢右上角 >>>点赞 | 收藏 <<<

你可能感兴趣的:(AI,人工智能)