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

相关推荐
曲幽1 分钟前
我用了FastApiAdmin后,连夜把踩过的坑都整理出来了
redis·python·postgresql·vue3·fastapi·web·sqlalchemy·admin·fastapiadmin
传说故事21 分钟前
【论文阅读】MotuBrain: An Advanced World Action Model for Robot Control
论文阅读·人工智能·具身智能·wam
北京耐用通信1 小时前
全域适配工业场景耐达讯自动化Modbus TCP 转 PROFIBUS 网关轻松实现以太网与现场总线互通
网络·人工智能·网络协议·自动化·信息与通信
火山引擎开发者社区1 小时前
TRAE × 火山引擎 Supabase:为你的 AI 应用装上“数据引擎”
人工智能
weixin_446260851 小时前
[特殊字符] 视觉Transformer (ViT) 原理及性能突破:从CNN到大规模自注意力机制的迁移
深度学习·cnn·transformer
小a彤1 小时前
GE 在 CANN 五层架构中的位置
人工智能·深度学习·transformer
前端若水1 小时前
会话管理:创建、切换、删除对话历史
前端·人工智能·python·react.js
Upsy-Daisy2 小时前
AI Agent 项目学习笔记(八):Tool Calling 工具调用机制总览
人工智能·笔记·学习
企学宝2 小时前
企学宝5月专题课程丨《OpenClaw AI 智能体实战营:从零基础部署到全场景自动化落地》
人工智能·ai·企业培训
碧海银沙音频科技研究院2 小时前
通话AEC与语音识别AEC的软硬回采链路
深度学习·算法·语音识别