-
查看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
相关推荐
独隅12 小时前
PyTorch 分布式训练完整指南:策略、实现与模型选型沪漂阿龙12 小时前
卷积神经网络(CNN)零基础通关指南:原理、图解与PyTorch实战龙文浩_1 天前
AI中NLP的注意力机制的计算公式解析断眉的派大星1 天前
pytorch中链式法则求解梯度龙文浩_2 天前
AI中NLP的深入浅出注意力机制 Seq2Seq 模型却道天凉_好个秋2 天前
pytorch(一):张量网上邻居YY2 天前
深度学习DL 之 安装PyTorch·GPU版、CUDA(本人Anaconda、Python、PyCharm已提前安装好)ZhiqianXia2 天前
Pytorch 学习笔记(21) : PyTorch Profiler盼小辉丶2 天前
PyTorch实战(41)——Hugging Face在PyTorch中的应用ZhiqianXia2 天前
PyTorch学习笔记(16):scheduler.py