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

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