-
查看python、pytorch、cuda以及显卡对应的版本
-
nvidia-smi查看显卡驱动,若报错则需要安装显卡驱动

- 这里的CUDA-Version意思是等下要安装的CUDA的版本要<=13.1
-
nvcc -V查看是否安装了cuda,若报错,还需要去安装cuda工具包,这一步要注意版本匹配

本地下载之后进行安装,一路默认即可。完成之后再次用
nvcc -V命令检查:

-
登录pytorch官网,进行选择对应的版本进行安装

-
检查是否安装成功。python ->>
import torch->>torch.is_available()

ptyorch安装
Forrit2026-02-09 23:20
相关推荐
田里的水稻1 小时前
FA_CNN卷积神经网络十八---感知机到多层感知机PyTorch听风吹等浪起2 小时前
第27章:EfficientNet全家桶|钢材表面6类缺陷分类完整Pytorch实现Ivanqhz3 小时前
MoE + LLM 应用tellmewhoisi1 天前
深度学习26:PyTorch:参数更新方法(Momentum)Zack_Liu1 天前
code样例论文复现现场1 天前
BEVFormer训练需要几张GPU?2张A100 80GB够不够,显存与云GPU成本分析Be for thing2 天前
PyTorch基础|Tensor张量详解,张量创建、数据类型与常用操作地平线开发者2 天前
QAT 实战:从 PyTorch 模型到 J6E 实板精度与延时验证tellmewhoisi2 天前
深度学习24:PyTorch:Sequential构建模型