安装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

说明安装成功

相关推荐
im_AMBER1 小时前
学习日志19 python
python·学习
白-胖-子2 小时前
深入剖析大模型在文本生成式 AI 产品架构中的核心地位
人工智能·架构
想要成为计算机高手4 小时前
11. isaacsim4.2教程-Transform 树与Odometry
人工智能·机器人·自动驾驶·ros·rviz·isaac sim·仿真环境
mortimer4 小时前
安装NVIDIA Parakeet时,我遇到的两个Pip“小插曲”
python·github
@昵称不存在4 小时前
Flask input 和datalist结合
后端·python·flask
静心问道4 小时前
InstructBLIP:通过指令微调迈向通用视觉-语言模型
人工智能·多模态·ai技术应用
宇称不守恒4.05 小时前
2025暑期—06神经网络-常见网络2
网络·人工智能·神经网络
赵英英俊5 小时前
Python day25
python
东林牧之5 小时前
Django+celery异步:拿来即用,可移植性高
后端·python·django
何双新5 小时前
基于Tornado的WebSocket实时聊天系统:从零到一构建与解析
python·websocket·tornado