Windows 下 CUDA, cudnn, Pytoch 卸载、更新、安装

hello, 大家好,今天趁着刚好要升级一下 cuda 来,用这个场景来演示一下 windows 下如何 CUDA, cudnn, pytoch 卸载、更新、安装。

安装准备

首先需要

查看本机支持的 cuda 版本及相关信息

这里有两种方法可以查看,效果是一样的。

我们先演示第一种

  • 通过 NVIDIA 控制面板查看

    • 桌面右键打开 NVIDIA 控制面板,点击 帮助-系统信息-组件 即可查看
查看本机支持的 cuda 版本
  • 通过 nvidia-smi 查看

    • 通过 cmd 打开控制台,输入命令查看
查看本机支持的 cuda 版本

这两种方法查看到的都是本机支持的最高版本的CUDA,它是可以向下兼容的。一般我们不需要安装最高版本,而是通过下面的结合 pytoch 版本来选择。

查看已安装版本

先来看看我已经安装的 CUDA 版本

  • nvcc -V
查看已安装版本

查看本机支持的 pytoch 版本 及相关信息

接着我们来看看本机支持的最高 pytoch 版本是多少,我们打开 pytorch 官网

PyTorch

这里我们可以看到本机,支持最高是 12.1 。

查看本机支持的 pytoch 版本

而我们上面查看到的本地实际上可以支持到 12.4,但是 pytorch 只支持到 12.1,所以我们 CUDA 只安装到 12.1 的版本就够了。

卸载旧版

然后我们继续卸载就版本的 CUDA

  • 控制面板设置-应用中卸载含 CUDA 字样的程序
卸载旧版

这里你们有什么卸载什么即可。

  • 进入到环境变量的 path 变量中,删除旧版本的环境变量

    • 同样是带 CUDA 字样

      我们这里检查了一下并没有,应该是卸载程序的时候系统自动帮我们删掉了,如果有请手动删除。

  • 进入 CUDA 目录,删除残留旧版本文件

    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

下载cuda安装包

接着我们去官网下载 CUDA

CUDA Toolkit Archive | NVIDIA Developer

我们这里要安装的是12.1 的版本。只需要关注小数点后一位即可,后面选最新即可。

选择好自己的平台,并点击下载

下载cuda安装包

这里的路径建议直接默认,避免很多麻烦。

这里提示我们有新版本可以忽略

我们选择自定义,因为我们只卸载了 CUDA 所以只需要安装 CUDA 即可。

安装 CUDA

这里的路径建议也是保持默认

安装 CUDA

然后下一步,等待安装即可。

安装注意:

如果你是第一次安装且没有安装 vs 的话,必须去掉在 CUDA 中的 visual studio integration 这一项。

但是如果有需要的请勾上。

检查

安装完成后,我们最好检查一下

  • 在命令行中输入nvcc -V查看安装的版本。
检查
  • 打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX\bin 检查nvcc.exe是否存在。

  • 打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX\extras\CUPTI\lib64 检查cupti64_XXX.dll是否存在。

下载cudnn安装包

接着我们继续下载 cudnn 库

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

它会叫你注册登录,你就注册登录即可。

找到适用于你自己 CUDA 的版本。

下载cudnn安装包

这里我下载过了,就直接演示

将下载到的压缩包解压到 cuda 的安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX下并覆盖。

检查环境变量

安装后重新检查环境变量的 path 变量中确认下面 3 项均存在,如果缺少则添加。

  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX\bin
  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX\libnvvp
  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXXX\extras\CUPTI\lib64

验证效果

随便打开个 python 环境,安装一下 pytoch

我们直接用官网的命令安装即可。

最后用下面的 python 代码来验证一下是否可以调用 GPU

import torch
torch.cuda.is_available() # 查看pytorch是否支持CUDA
torch.cuda.device_count() # 查看可用的CUDA数量
torch.version.cuda # 查看对应CUDA的版本号
验证效果

至此,我们的卸载,更新,安装就教完了。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,835评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,598评论 1 295
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,569评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,159评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,533评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,710评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,923评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,674评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,421评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,622评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,115评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,428评论 2 254
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,114评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,097评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,875评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,753评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,649评论 2 271

推荐阅读更多精彩内容