安装GPU版本的Pytorch

前言

Pytorch是深度学习框架,在工作中我们一般是使用GPU版本的Pytorch,提高运行效率

安装GPU版本的Pytorch需要先安装CUDA和CUANN这两个GPU环境

如果准备安装GPU版本的Pytorch安装同志没有安装CUDA和CUANN,请看我上一篇文章

RTX5070显卡安装CUDA和CUDNN-CSDN博客

目录

安装GPU版本的Pytorch

获取下载pytorch的命令

创建虚拟环境

执行在pytorch官网获得的命令

验证GPU版本的Pytorch安装是否成功


安装GPU版本的Pytorch

获取下载pytorch的命令

pytorch官网

Get Started

选择适版本匹配的pytorch

"Stable"表示稳定版本(通常推荐使用稳定版)

"Your OS"选择操作系统,如 Windows

"Package"选择包管理工具,如 pip 或 conda

"Language"选择 构建语言,选择python

"Compute Platform"选择对应的CUDA版本,这里我选择12.8。

拿到pytorch的安装命令

创建虚拟环境

在开始菜单里面搜索prompt找到Anaconda prompt命令窗口打开

创建conda虚拟环境并且指定这个虚拟环境的python版本

首选查看你的Anaconda支持的python版本

conda search python

找到想要指定的python版本

创建虚拟环境

自定义虚拟环境的默认路径(可选)

我指定的是D:\environment\AnacondaEnvs文件夹管理我的虚拟环境

conda config --add envs_dirs 准备存放虚拟环境的目录

创建虚拟环境

conda create --name 虚拟环境名称 python=版本号

我创建的虚拟环境的名称是pytorch,指定python版本是3.13.1

我的Anaconda是支持3.13.1版本的,看自己Anaconda支持python版本的范围自己决定

查看刚才创建的虚拟环境

conda env list

进入到刚才创建的虚拟环境中,就是激活虚拟环境

conda activate 安装虚拟环境的路径(最后一个文件夹是虚拟环境的名称)

我这里进入我刚才创建的pytorch虚拟环境

执行在pytorch官网获得的命令

执行成功之后退出虚拟环境

conda deactivate

验证GPU版本的Pytorch安装是否成功

使用pyCharam编译器指定pytorch虚拟环境创建一个项目

创建项目名为MyPytorch,指定pytorch虚拟环境

进入项目编写这几行代码

复制代码
import torch

print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

torch.version 是查看pytorch版本

torch.version.cuda 是查看pytorch使用的cuda版本

torch.cuda.is_available() 是判断是否使用的是GPU版本 True是 False否

torch.cuda.get_device_name(0) 是查看pytorch使用的GPU版本

如果运行正常。 torch.cuda.is_available()是True

说明安装成功

相关推荐
美狐美颜sdk39 分钟前
跨平台直播美颜SDK集成实录:Android/iOS如何适配贴纸功能
android·人工智能·ios·架构·音视频·美颜sdk·第三方美颜sdk
DeepSeek-大模型系统教程1 小时前
推荐 7 个本周 yyds 的 GitHub 项目。
人工智能·ai·语言模型·大模型·github·ai大模型·大模型学习
郭庆汝1 小时前
pytorch、torchvision与python版本对应关系
人工智能·pytorch·python
小雷FansUnion3 小时前
深入理解MCP架构:智能服务编排、上下文管理与动态路由实战
人工智能·架构·大模型·mcp
资讯分享周3 小时前
扣子空间PPT生产力升级:AI智能生成与多模态创作新时代
人工智能·powerpoint
思则变4 小时前
[Pytest] [Part 2]增加 log功能
开发语言·python·pytest
叶子爱分享4 小时前
计算机视觉与图像处理的关系
图像处理·人工智能·计算机视觉
鱼摆摆拜拜4 小时前
第 3 章:神经网络如何学习
人工智能·神经网络·学习
一只鹿鹿鹿4 小时前
信息化项目验收,软件工程评审和检查表单
大数据·人工智能·后端·智慧城市·软件工程
张较瘦_5 小时前
[论文阅读] 人工智能 | 深度学习系统崩溃恢复新方案:DaiFu框架的原位修复技术
论文阅读·人工智能·深度学习