-
查看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
相关推荐
Rorsion8 小时前
PyTorch实现二分类（单特征输出+单层神经网络）不懒不懒16 小时前
【从零开始：PyTorch实现MNIST手写数字识别全流程解析】工程师老罗1 天前
基于Pytorch的YOLOv1 的网络结构代码JarryStudy1 天前
HCCL与PyTorch集成 hccl_comm.cpp DDP后端注册全流程Eloudy1 天前
用 Python 直写 CUDA Kernel的技术，CuTile、TileLang、Triton 与 PyTorch 的深度融合实践Rorsion2 天前
PyTorch实现线性回归骇城迷影2 天前
Makemore 核心面试题大汇总mailangduoduo2 天前
零基础教学连接远程服务器部署项目——VScode版本多恩Stone2 天前
【3D AICG 系列-6】OmniPart 训练流程梳理