深度学习(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版本安装成功。

相关推荐
金井PRATHAMA7 分钟前
框架系统的多维赋能——论其对自然语言处理深层语义分析的影响与启示
人工智能·自然语言处理·知识图谱
面壁的熊猫9 分钟前
目标检测概述
人工智能·目标检测·计算机视觉
Learn Beyond Limits15 分钟前
Using per-item Features|使用每项特征
人工智能·python·神经网络·算法·机器学习·ai·吴恩达
石臻臻的杂货铺22 分钟前
如何让AI实现自动化 —— PlayWright MCP 实测
运维·人工智能·自动化
之墨_23 分钟前
【大语言模型】—— Transformer的QKV及多头注意力机制图解解析
人工智能·语言模型·transformer
我的offer在哪里31 分钟前
九月技术奇点观察:当量子算力与 AI 认知同时突破临界点
人工智能·量子计算
文火冰糖的硅基工坊40 分钟前
[创业之路-682]:实即虚,虚即实。真正的技术壁垒,藏在光路之外、电路之下、代码之中。
人工智能·算法·系统架构·制造·创业·产业链
nju_spy1 小时前
大模型面经(一) Prompt + RAG + 微调
人工智能·面试·lora·大模型·rag·提示词工程·peft微调
学技术的大胜嗷1 小时前
使用BatchNorm偏置填充边界:确保推理一致性与数值稳定性
人工智能·深度学习·图像生成
玩转AGI1 小时前
【必收藏】12-Factor Agents:让大模型Agent从能跑起来到能用起来的企业级设计指南
人工智能·程序员·llm