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

相关推荐
java1234_小锋7 分钟前
[免费]基于Python的协同过滤电影推荐系统(Django+Vue+sqlite+爬虫)【论文+源码+SQL脚本】
python·django·电影推荐系统·协同过滤
看海天一色听风起雨落1 小时前
Python学习之装饰器
开发语言·python·学习
小憩-1 小时前
【机器学习】吴恩达机器学习笔记
人工智能·笔记·机器学习
却道天凉_好个秋1 小时前
深度学习(二):神经元与神经网络
人工智能·神经网络·计算机视觉·神经元
UQI-LIUWJ1 小时前
unsloth笔记:运行&微调 gemma
人工智能·笔记·深度学习
XiaoMu_0011 小时前
基于Python+Streamlit的旅游数据分析与预测系统:从数据可视化到机器学习预测的完整实现
python·信息可视化·旅游
THMAIL1 小时前
深度学习从入门到精通 - 生成对抗网络(GAN)实战:创造逼真图像的魔法艺术
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·cnn
却道天凉_好个秋1 小时前
计算机视觉(八):开运算和闭运算
人工智能·计算机视觉·开运算与闭运算
无风听海1 小时前
神经网络之深入理解偏置
人工智能·神经网络·机器学习·偏置
JoinApper1 小时前
目标检测系列-Yolov5下载及运行
人工智能·yolo·目标检测