ptyorch安装

  1. 查看python、pytorch、cuda以及显卡对应的版本

  2. nvidia-smi查看显卡驱动,若报错则需要安装显卡驱动

    • 这里的CUDA-Version意思是等下要安装的CUDA的版本要<=13.1
  3. nvcc -V查看是否安装了cuda,若报错,还需要去安装cuda工具包,这一步要注意版本匹配
    本地下载之后进行安装,一路默认即可。

    完成之后再次用nvcc -V命令检查:

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

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

相关推荐
盼小辉丶4 天前
PyTorch实战(30)——使用TorchScript和ONNX导出通用PyTorch模型
人工智能·pytorch·深度学习·模型部署
封奚泽优4 天前
使用mmdetection项目进行训练记录
pytorch·python·cuda·mmdetection·mmcv
tony3654 天前
pytorch分布式训练解释
人工智能·pytorch·分布式
weixin_贾4 天前
深度学习基础理论与 PyTorch 实战 —— 从传统机器学习到前沿模型全攻略
pytorch·深度学习·机器学习
大连好光景5 天前
PyTorch深度学习----优化器
pytorch·深度学习·学习
多恩Stone6 天前
【3D-AICG 系列-11】Trellis 2 的 Shape VAE 训练流程梳理
人工智能·pytorch·算法·3d·aigc
隔壁大炮6 天前
08. PyTorch_张量基本创建方式
人工智能·pytorch·python
隔壁大炮6 天前
07. PyTorch框架简介
人工智能·pytorch·python
大鹏的NLP博客6 天前
Rust + PyTorch 实现 BGE 向量检索系统
人工智能·pytorch·rust
勾股导航8 天前
蚁群优化算法
人工智能·pytorch·python