人工智能:深度学习:0.pytorch安装

一、查看 CUDA 版本(GPU 环境必做,无 GPU 可跳过)

  1. 打开开始菜单栏,搜索PowerShell ,点击打开第一个结果;
  2. 在终端中输入命令:nvidia-smi,回车执行;
  3. 从输出结果中查看显卡驱动支持的 CUDA 版本(示例为 12.7)。

二、下载安装 CUDA Toolkit

  1. 进入 CUDA 官方下载页:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local;
  2. 根据自身电脑配置选择对应参数(示例为 Windows11 系统),安装方式选择exe(local) 下载本地安装包;
  3. 下载完成后,双击安装包按提示完成 CUDA Toolkit 安装。

三、获取 PyTorch 官方安装命令

  1. 进入 PyTorch 官方安装页:https://pytorch.org/get-started/locally/;
  2. 因官网暂未提供 12.7 版本,选择低于 12.7 的兼容版本 12.6;
  3. 按自身环境依次选择:系统(Windows)→ 包管理工具(Pip),页面将自动生成适配的安装命令,复制备用。

四、执行 PyTorch 安装命令

  1. 打开 PyCharm,点击软件左下角的终端 面板;
  2. 将复制好的官方安装命令粘贴至终端,回车执行(因为我已经安装,所以并没有下载过程,全新安装将自动完成依赖包下载与配置)。

五、验证安装是否成功

在 PyCharm 终端中输入python进入 Python 交互环境,依次执行以下代码:

python

运行

python 复制代码
import torch
# 查看PyTorch安装版本
print(torch.__version__)
# 验证GPU加速是否可用
print(torch.cuda.is_available())

若成功打印 PyTorch 版本号,即表示安装成功;GPU 环境下torch.cuda.is_available()返回True,代表 GPU 加速可正常使用。

相关推荐
程序人生888几秒前
从“把隐私文件上传到别人云端“到本地可控:DocConverter Web 立项初心与架构选型
前端·图像处理·人工智能·opencv·架构·ocr·文心一言
数智工坊几秒前
视觉SLAM第11讲|回环检测:词袋模型、字典构建与相似度计算全解析
人工智能·深度学习·线性代数
TechEdu2026062 分钟前
[人工智能]Python11:NumPy 源代码、软件架构与软件流程
人工智能·numpy
loulanyue_3 分钟前
突破单点AI瓶颈:慧博零售全域Agent的数智化实践——读慧博科技CTO贾世龙2026云栖专场演讲
人工智能·科技·零售
蜗牛互联网7 分钟前
Java HttpClient 调用 Gemini 图像理解与金额字段校验
java·开发语言·人工智能·后端·python
开开心心就好7 分钟前
视频里的图片怎么提取?双击一下就导出
java·前端·人工智能·spring·智能手机·intellij-idea·excel
蜗牛互联网8 分钟前
Python Responses API视觉输入与本地金额校验最小实现
java·开发语言·人工智能·后端·python
Autumn_ing9 分钟前
2026产品经理Agent实测:WorkBuddy、QoderWork、Codex、墨刀AI客户端
人工智能·产品经理·墨刀·codex·workbuddy·qoderwork
Mr_star_galaxy10 分钟前
【Agent】LangChain聊天模型 -- 调用工具
人工智能·langchain
海宇数据10 分钟前
零信任架构实战:基于海宇车辆vin码查车辆信息详版构建自动化车险承保定级网关
人工智能·架构·自动化·php