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 小时前
GPU 编号进阶:CUDA\_VISIBLE\_DEVICES、多进程与容器化陷阱
人工智能·pytorch·python
吴佳浩4 小时前
GPU 编号错乱踩坑指南:PyTorch cuda 编号与 nvidia-smi 不一致
人工智能·pytorch·nvidia
云上的云端8 小时前
vLLM-Ascend operator torchvision::nms does not exist 问题解决
人工智能·pytorch·深度学习
Zhansiqi8 小时前
dayy43
pytorch·python·深度学习
写点什么呢19 小时前
Pytorch学习16_损失函数与反向传播
人工智能·pytorch·python·学习·pycharm
罗罗攀1 天前
PyTorch学习笔记|张量的广播和科学运算
人工智能·pytorch·笔记·python·学习
IT阳晨。2 天前
PyTorch深度学习实践
人工智能·pytorch·深度学习
Sakuraba Ema2 天前
从零理解 MoE(Mixture of Experts)混合专家:原理、数学、稀疏性、专家数量影响与手写 PyTorch 实现
人工智能·pytorch·python·深度学习·数学·llm·latex
Westward-sun.2 天前
PyTorch入门实战:MNIST手写数字识别(全连接神经网络详解)
人工智能·pytorch·神经网络
weixin_457760002 天前
基于pytorch实现LPR模型车牌识别
人工智能·pytorch·python·深度学习·lpr