显卡驱动

到这里下载显卡驱动更新工具，该工具负责安装最新的驱动及更新

查看可以安装cuda版本

nvidia-smi

在这里插入图片描述
例如，我这里可以安装的cuda版本为：12.1

下载cuda并安装

下载

点击这里进入下载页面：
在这里插入图片描述

这里下载 12.1 大版本即可

安装

在这里插入图片描述

取消这个选项：

在这里插入图片描述
安装路径看情况是否使用默认路径，只要以下环境变量配置正确：
在这里插入图片描述

安装cudnn

  • 点击这里下载
  • 将压缩包解压，然后把包含的3个文件夹复制到CUDA安装路径下即可

最新：其实安装pytorch-gpu版，不需要单独安装上面的CUDA和cuDNN，torch的安装包集成的有这俩玩意，你只需要保证你的N卡的驱动版本号高于torch的最低要求。直接参考下面安装pytorch

安装pytorch

进入官网，在你的解释器运行下图红框的命令：
在这里插入图片描述

验证

import torch
torch.cuda.is_available()
# windows # python # pytorch
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

Datawhale组队学习-【AI4S】公开课 Lesson 1：流体场自编码器

这张图里总结了通用的好的表征学习表征问题往往先于模型问题表征决定模型上限好的表征学习，就是把复杂的原始数据（图像、分子、文本等）映射到一个**既保持了原始结构（距离保持、连续性），又方便我们操作（可插值、可加性）**的向量空间，让后续的机器学习任务（分类、生成、预测）变得更容易、更鲁棒。两种典型的 “坏” 情况情况一：距离高估（Distance Overestimate）**现象：**两张几乎一样

avatar 2048 AI社区

掌握AI原生应用领域持续学习，引领行业潮流

在当今科技飞速发展的时代，AI原生应用正逐渐改变着各个行业的面貌。本文的目的就是帮助大家了解AI原生应用领域的知识，掌握持续学习的方法，从而能够在这个领域中紧跟潮流，甚至引领潮流。我们的讨论范围涵盖了AI原生应用的核心概念、算法原理、实际应用等多个方面。本文将首先介绍AI原生应用领域的核心概念，让大家对这个领域有一个初步的认识。然后详细讲解核心算法原理和数学模型，通过实际的代码案例帮助大家理解和应

avatar 2048 AI社区

LangChain 检索增强生成（RAG）

检索增强生成（Retrieval-Augmented Generation, RAG） 是一种将信息检索与语言模型生成相结合的技术范式。其核心思想是：在语言模型生成答案之前，先从外部知识库中检索出与问题相关的文档片段，将这些片段作为上下文注入模型，从而让模型能够利用最新、最专业或私有数据生成更准确、更可信的回答。

avatar 2048 AI社区