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()

相关推荐
一次旅行21 小时前
Attention机制从数学到工程:拆解缩放点积+多头注意力|附可运行PyTorch实现与踩坑指南
人工智能·pytorch·python
ai小陈1 天前
PyTorch多GPU分布式训练实战:从单卡脚本迁移到DDP
服务器·人工智能·pytorch·分布式·深度学习·ai·gpu算力
EW Frontier1 天前
CMA-Former:基于级联多头注意力 Transformer 的 DOA 波达方向估计(PyTorch 实现)
pytorch·深度学习·transformer·doa估计
wangchen_02 天前
PyTorch
人工智能·pytorch·python
%473 天前
DAY41
pytorch·python
一只积极向上的小咸鱼3 天前
pytorch 与资源核算
人工智能·pytorch·python
小女孩真可爱3 天前
GPT(3)----------------GQA分组查询注意力机制提速
人工智能·pytorch·gpt·深度学习·大模型
Three_ST3 天前
沐神-动手学习深度学习-习题 4.3. 多层感知机的简洁实现
人工智能·pytorch·python·深度学习·机器学习
Zzj_tju3 天前
复现一个 Transformer Encoder:从论文公式到最小 PyTorch
人工智能·pytorch·深度学习·语言模型
小桥流水---人工智能3 天前
Windows下RTX 5070安装PyTorch GPU完整教程:Python 3.11 + PyTorch 2.8.0 + CUDA 12.9
pytorch·windows·python3.11