深度学习(11)--PyTorch(GPU)安装

一.安装Anaconda

Anaconda提供了一个包管理系统和环境管理工具,使用户可以方便的安装、管理和切换不同版本的Python以及相关数据包。

具体安装步骤可以参考博主的文章:

深度学习(4)--Keras安装-CSDN博客https://blog.csdn.net/GodFishhh/article/details/135749563?spm=1001.2014.3001.5501安装完Anaconda后,打开Anaconda Prompt

创建一个虚拟环境用以存放PyTorch环境:

复制代码
conda create -n pytorch-gpu python=3.9

虚拟环境的名字pytorch-gpu和python版本可以根据自己的需求设置。

使用conda activate pytorch-gpu尝试激活创建的虚拟环境,如果激活成功则正确创建虚拟环境:

查看环境内安装的python版本:

二.安装PyTorch

查看电脑对应的cuda版本

博主可安装CUDA的最高版本为12.2

在pytorch官网找到对应版本的安装包
https://pytorch.org/https://pytorch.org/

进入安装好的pytorch-gpu环境,输入上述Run this Command中的下载指令

检测是否安装成功:

显示输出为true,则说明PyTorch的GPU版本安装成功。

相关推荐
kaixin_啊啊4 小时前
【零基础学AI】第 24 章课后练习与答案
人工智能
EchoMind-Henry4 小时前
娱乐资本论发布AI短剧剧本对比文章
人工智能·娱乐
谁在黄金彼岸4 小时前
Lightpanda无头浏览器实战-Playwright接入与Agent化
人工智能
hkNaruto4 小时前
【AI】游戏大模型智能体OODA过程记忆问题解决方案:AgenticSTS 有界记忆契约:技术分析与流水线改造操作手册
人工智能·游戏·机器学习
杜子不疼.4 小时前
没有真机,我给启元 Q1 造了一台“假机器人“:官方 SDK 示例跑通实录
人工智能·机器人
qq_280391494 小时前
温度参数设置建议
人工智能
qq_280391495 小时前
Java构建飞书消息格式详解
人工智能
七元权5 小时前
FoundationPose 推理全流程解读:从一帧 RGB-D 到 6D 位姿
论文阅读·深度学习·6d姿态估计
Flynn Liang5 小时前
上线后怎么排查 AI 调用?Spring AI Observability:token/延迟/成本埋点与全链路追踪
java·人工智能·spring·监控·可观测性·observability·springai
木头左5 小时前
【Python】不点浏览器发博客:逆向 CSDN 创作中心接口做 md-publish 技能
python·自动化·api