当前位置: 首页 > news >正文

【PyTorch】2024保姆级安装教程-Python-(CPU+GPU详细完整版)-

PyTorch 安装指南(CPU/GPU 版本)

环境准备

确保已安装 Python(推荐 3.8-3.10 版本)和 pip 包管理工具。可通过以下命令验证:

python --version pip --version

对于 GPU 版本,需提前安装 CUDA 工具包(推荐 CUDA 11.8)和 cuDNN。通过以下命令检查 CUDA 是否可用:

nvidia-smi
安装 PyTorch

访问 PyTorch 官网获取最新安装命令(https://pytorch.org/get-started/locally/),或直接使用以下通用命令:

CPU 版本

pip install torch torchvision torchaudio

https://www.zhihu.com/zvideo/1994565209331885213/
https://www.zhihu.com/zvideo/1994565208023258822/
https://www.zhihu.com/zvideo/1994565207540925794/
https://www.zhihu.com/zvideo/1994565206941127434/
https://www.zhihu.com/zvideo/1994565206744003950/
https://www.zhihu.com/zvideo/1994565206433625958/
https://www.zhihu.com/zvideo/1994565206098077303/
https://www.zhihu.com/zvideo/1994565204680397752/
https://www.zhihu.com/zvideo/1994565203636003581/
https://www.zhihu.com/zvideo/1994565201907955097/
https://www.zhihu.com/zvideo/1994565201689875180/
https://www.zhihu.com/zvideo/1994565200905524719/
https://www.zhihu.com/zvideo/1994565200305737977/
https://www.zhihu.com/zvideo/1994565198812561760/
https://www.zhihu.com/zvideo/1994565198850327780/
https://www.zhihu.com/zvideo/1994565197306815368/
https://www.zhihu.com/zvideo/1994565196514103586/
https://www.zhihu.com/zvideo/1994565196014966417/
https://www.zhihu.com/zvideo/1994565190486890060/
https://www.zhihu.com/zvideo/1994565189287318935/
https://www.zhihu.com/zvideo/1994565187051726155/
https://www.zhihu.com/zvideo/1994565188213569352/
https://www.zhihu.com/zvideo/1994565185751512843/
https://www.zhihu.com/zvideo/1994565186321921600/
https://www.zhihu.com/zvideo/1994565184279298762/
https://www.zhihu.com/zvideo/1994565184161858093/
https://www.zhihu.com/zvideo/1994565183075526026/
https://www.zhihu.com/zvideo/1994565181632692694/
https://www.zhihu.com/zvideo/1994565181045486231/
https://www.zhihu.com/zvideo/1994565181108413045/
https://www.zhihu.com/zvideo/1994565180231812002/
https://www.zhihu.com/zvideo/1994565179363571467/
https://www.zhihu.com/zvideo/1994565177887199780/
https://www.zhihu.com/zvideo/1994565177253855407/
https://www.zhihu.com/zvideo/1994565176838595115/
https://www.zhihu.com/zvideo/1994565175676777824/
https://www.zhihu.com/zvideo/1994565174481404230/
https://www.zhihu.com/zvideo/1994565174229750335/
https://www.zhihu.com/zvideo/1994565171113375318/
https://www.zhihu.com/zvideo/1994565168638742902/
https://www.zhihu.com/zvideo/1994565169397912751/
https://www.zhihu.com/zvideo/1994565168420638729/
https://www.zhihu.com/zvideo/1994565167074268258/
https://www.zhihu.com/zvideo/1994565168265442742/
https://www.zhihu.com/zvideo/1994565167678261233/
https://www.zhihu.com/zvideo/1994565166956823859/
https://www.zhihu.com/zvideo/1994565166822618564/
https://www.zhihu.com/zvideo/1994565166868763135/
https://www.zhihu.com/zvideo/1994565165560140329/

GPU(CUDA 11.8)版本

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
验证安装

运行 Python 解释器执行以下代码:

import torch print(torch.__version__) # 查看版本 print(torch.cuda.is_available()) # GPU 是否可用
常见问题解决
  • CUDA 不匹配:卸载现有版本后重新安装指定 CUDA 版本的 PyTorch。
  • 权限问题:添加--user参数或使用虚拟环境。
  • 下载慢:更换 pip 源(如-i https://pypi.tuna.tsinghua.edu.cn/simple)。
虚拟环境推荐

使用 conda 或 venv 创建独立环境:

conda create -n pytorch_env python=3.9 conda activate pytorch_env
http://www.jsqmd.com/news/240898/

相关文章:

  • 资深Android开发工程师职位深度解析:基于广州极飞科技股份有限公司的职位需求
  • CCS20高可用性架构构建:实际案例分享
  • Anthropic 重磅发布 Cowork:让普通人都能用上Claude Code!
  • 全网最全9个一键生成论文工具,本科生毕业论文必备!
  • Flink:窗口同组联结(Window CoGroup)
  • 意料之内的回调,倒车接人?
  • Figma中文界面终极指南:3步解锁专业设计无障碍体验
  • 【顶级EI复现】基于主从博弈的售电商多元零售套餐设计与多级市场购电策略(Matlab代码实现)
  • 人工智能之数学基础:正态分布在机器学习领域中的体现
  • STM32CubeMX安装兼容性问题解决:全面讲解方案
  • USB-Blaster驱动手动安装步骤:系统学习与实践
  • 软件测试面试(内附答案)
  • 厦门大学突破:多模态模型进入“推理进化”时代,7B小模型的大逆袭
  • 【复现】MFAC无模型自适应控制CFDL+PFDL+FFDL三个非线性系统动态线性研究(Matlab代码实现)
  • Java 提供了丰富的数据结构来处理和组织数据【1】
  • Keil uVision5嵌入式C开发:新手教程(从零配置环境)
  • 【倒计时一天】2025第八届金猿大数据产业发展论坛——暨AI InfraData Agent趋势论坛丨颁奖典礼·上海
  • 搞定NPU推理加速实战
  • 智能穿戴OLED显示方案:SSD1306中文手册系统学习
  • 了解Java 数据结构【1】
  • Matlab实现图正则化稀疏编码(GraphSC)算法详解
  • javascript数据类型转换-转换为数字型
  • 基于实际项目的PCB布局布线思路:初级应用示范
  • 批量 roi 目录 roi
  • 项目应用中c++ spidev0.0 read值为255的解决方案
  • 麒麟V10-ARM架构Docker启动报错
  • LVGL移植实战案例:配合DMA2D加速GUI绘制
  • 三种神经网络BP-PID、RBF-PID、单神经元自适应优化PID算法对比仿真(程序+参考资料)
  • 零基础入门:理解AUTOSAR中DIO驱动配置
  • Cortex-M浮点单元(FPU)使用指南:新手必看示例