pyotrch GPU版本 安装

author：徐昊

pytorch安装

地址Start Locally | PyTorch

1.CPU版本

pip安装默认cpu版本

2.GPU版本

1.显卡算力、显卡驱动、cuda版本

三者之间需要相互匹配

显卡安装的时候就会安装cuda driver（让显卡能进行并行计算）

然后安装cuda runtime 去调用cuda driver 去让显卡进行并行运算

cuda runtime 版本最好要低于cuda driver （驱动版本要高于软件版本）

2.显卡型号确认查询算力

打开任务管理器

算力查看https://en.wikipedia.org/wiki/CUDA 3050算力8.6 架构 Ampere

40系列8.9 30系列8.6

3.根据算力去选择cuda版本

3050 对应 CUDA SDK 版本是 11.1-11.4

能满足CUDA Driver 11.0-12.5

所以CUDA Runtime 可以选择低于12.5的全部

4.查看自己的驱动CUDA Driver Version

win+r 输入cmd

nvidia-smi

驱动更新英伟达可以去GeForce

5.安装显卡驱动最新版本

nvidia-smi
##如果显示没有这条命令 就去安装 有反馈就看自己想不想

1.地址https://www.nvidia.cn/drivers/lookup/

选择驱动下载安装

2.GeForce Experience

6.conda 环境安装pytorch

安装pytorch (1.pytorch 2.torchvision图片处理工具 3.torchaudio语言处理)

我的cuda版本为12.6但是pythorch最高支持是12.4 （12.6肯定足够，向下兼容）

stable稳定版 preview预览版

速度非常慢非常不推荐

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

推荐安装方式

先选择pip

pip3 install torch --index-url https://download.pytorch.org/whl/cu124

获取包地址

使用第三方软件下载

pip 本地安装

再安装剩余两个包

pip3 install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

查看安装的包

7.验证是否安装成功

python
import torch
torch.cuda.is_available()
##显示true 则为安装成功