【操作与配置】Pytorch环境搭建

安装显卡驱动

显卡驱动是一种软件程序,用于控制显卡硬件与操作系统之间的通信和交互。显卡驱动负责向操作系统提供有关显卡硬件的信息,以及使操作系统能够正确地控制和管理显卡的各种功能和性能。显卡驱动还包含了针对不同应用程序和游戏的优化,以提高显卡性能和图形渲染效果。更新和安装最新的显卡驱动可以提高计算机的图形性能和稳定性。

NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

在官网获取你的显卡驱动,版本较低的也可下载更新程序

安装CUDA

CUDA(Compute Unified Device Architecture)是NVIDIA推出的一种并行计算平台和编程模型,可以利用GPU的并行计算能力加速各种应用程序,包括机器学习、图像处理、科学计算等。

Microsoft Windows 的 CUDA 安装指南 (nvidia.com)

可参照官方安装指南

CUDA Toolkit 12.5 Downloads | NVIDIA Developer

在此处下载CUDA Toolkit

安装cuDNN

CuDNN(CUDA Deep Neural Network)是NVIDIA为深度学习框架提供的一个GPU加速库,包括大量优化的深度学习算法和函数,能够帮助开发者在GPU上快速实现高效的深度学习模型训练和推理。CuDNN主要用于神经网络的加速计算。

现行Pytorch版本疑似内置cuDNN,无需下载

在官网按照配置下载

将解压缩的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5

安装Anaconda

【操作与配置】Python:Conda+Pycharm_conda pycharm csdn-CSDN博客

安装PyTorch

PyTorch是一个开源的深度学习框架,由Facebook的人工智能研究小组开发。它提供了一个灵活的、动态的计算图的机制,使得深度学习模型的设计和训练变得更加直观和灵活。PyTorch使用Python作为开发语言,与Python的科学计算库NumPy和科学计算库SciPy深度整合,方便用户进行数据处理和模型训练。

PyTorch支持动态计算图和静态计算图两种方式进行模型构建,用户可以根据自己的需求选择适合的方式。它还提供了大量方便的工具和库,用于构建神经网络、优化模型、处理数据等。PyTorch在学术界和工业界都有着广泛的应用,被认为是一个功能强大、易用的深度学习框架

查询你所需的conda命令,在conda中使用

Start Locally | PyTorch

运行代码查看pytorch版本以及cuda是否可用

python 复制代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
相关推荐
珠海西格电力1 分钟前
零碳园区产业园管理系统的全场景源网荷储氢协同调度功能是如何实现的
大数据·运维·人工智能·物联网·能源
smallyoung2 分钟前
具有反思能力的 Agentic RAG 实战:用 LangChain4j 实现 CRAG 纠错检索
人工智能·后端
wenzhangli75 分钟前
Harness Engineering:AICode 的灵魂——Ooder A2UI 从难产到重生的深度实践
人工智能·ai编程
Deepoch9 分钟前
Deepoc 具身模型开发板在田间除草机器人自主作业中的技术应用
人工智能·机器人·具身模型·deepoc·除草机器人
Muyuan199811 分钟前
22.让 RAG Agent 更像真实产品:聊天页面优化、PDF 上传、知识库重建与检索片段展示
python·django·pdf·fastapi
ai大模型中转api测评15 分钟前
解密 GPT-5.5:原生多模态架构如何重定义 AI 逻辑推理与精准制图
大数据·人工智能·gpt·架构·api
冷雨夜中漫步18 分钟前
Claude Code源码分析——Claude Code Agent Loop 详细设计文档
java·开发语言·人工智能·ai
xixixi7777722 分钟前
英伟达Agent专用全模态模型出击,仿冒AI智能体泛滥成灾,《AI伦理安全指引》即将落地——AI治理迎来“技术-风险-规范”三重奏
人工智能·5g·安全·ai·大模型·英伟达·智能体
程序员-小李22 分钟前
uv 学习总结:从零到一掌握现代化 Python 工具链
python·学习·uv
直奔標竿24 分钟前
Java开发者AI转型第二十六课!Spring AI 个人知识库实战(五)——联网搜索增强实战
java·开发语言·人工智能·spring boot·后端·spring