author:徐昊
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 则为安装成功