-
查看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
相关推荐
pythonpioneer10 小时前
PyTorch3D:基于 PyTorch 的高效 3D 深度学习工具库python-码博士15 小时前
PyTorch 从零实现 Flow Matching:训练、采样、画图一条龙努力写A题的小菜鸡15 小时前
PyTorch 图像预处理 transforms 与 TensorBoard 可视化 (自己学习记录)装不满的克莱因瓶16 小时前
自然语言处理常见任务——从文本理解到生成式AI的完整任务体系装不满的克莱因瓶19 小时前
自然语言处理中的分词——从语言切分到模型输入的第一步All The Way North-21 小时前
大模型训练必修课:梯度裁剪(Gradient Clipping)从数学原理,到PyTorch工程实战全解析zzzzzz3101 天前
LMCache 深度解析:LLM 推理加速的秘密武器,TTFT 降低 13 倍是怎么做到的?装不满的克莱因瓶1 天前
掌握条件生成对抗网络(Conditional GAN)模型结构——从无条件生成到可控生成的进阶丨白色风车丨1 天前
PyTorch 实现手写数字识别:全连接网络 + CNN 卷积网络(MNIST 数据集实战)装不满的克莱因瓶1 天前
掌握生成对抗网络(GAN)原理——从零理解“对抗学习”的核心思想与生成机制