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

相关推荐
笑小枫12 分钟前
Pytorch使用GPU训练全过程,包含安装CUDA、cuDNN、PyTorch
人工智能·pytorch·python
Blossom.11828 分钟前
深度学习中的注意力机制:原理、应用与实践
人工智能·深度学习·神经网络·机器学习·生成对抗网络·计算机视觉·sklearn
【本人】29 分钟前
Django基础(二)———URL与映射
后端·python·django
飞哥数智坊1 小时前
Cursor替代方案整理,附模型不可用进阶解决方案
人工智能·claude·cursor
摸鱼仙人~1 小时前
现代人工智能综合分类:大模型时代的架构、模态与生态系统
人工智能·分类·数据挖掘
麻雀无能为力1 小时前
CAU数据挖掘第四章 分类问题
人工智能·分类·数据挖掘·中国农业大学计算机
mit6.8241 小时前
[AI-video] 数据模型与架构 | LLM集成
开发语言·人工智能·python·微服务
roman_日积跬步-终至千里1 小时前
【机器学习【6】】数据理解:数据导入、数据审查与数据可视化方法论
人工智能·机器学习
lucky_lyovo1 小时前
卷积神经网络-卷积的分类
深度学习·分类·cnn
蓝婷儿1 小时前
Python 数据建模与分析项目实战预备 Day 4 - EDA(探索性数据分析)与可视化
开发语言·python·数据分析