当前位置: 首页 > news >正文

【Pytorch】Pytorch或者CUDA版本不符合问题解决与分析

NVIDIA CUDA Toolkit Release Notes

Package installation issues

INSTALL PYTORCH

先声毒人:最好资料就是上面三份资料,可以通过官网明确的获取一手信息,你所遇到的99%的问题都可以找到,明确的解决方案,建议最好看原材料,避免走弯路,卡壳!!!!!

【Pytorch】Pytorch或者CUDA版本不符合问题解决与分析

      • 1. 版本不符问题解决与分析
      • 2. 【Pytorch】Pytorch或者CUDA版本不符合问题解决

1. 版本不符问题解决与分析

导致这一问题的原因主要分为如下四个方面,在这里先作总结,具体方法在之后展开。有经验的同学可以直接根据本节进行快速排查,以提高效率。

  1. CUDA版本与驱动程序不兼容:CUDA版本需与GPU驱动程序兼容。
  2. CUDA库的路径设置存在问题:如果CUDA库路径未正确配置,PyTorch将无法找到CUDA库文件。
  3. PyTorch版本与CUDA版本不匹配:安装的PyTorch版本需与CUDA版本匹配。
  4. 编译问题:如果使用预编译的PyTorch二进制文件,可能存在与CUDA版本不匹配的问题。
  5. 冲突:可能存在其他软件包或库与PyTorch或CUDA发生冲突。

2. 【Pytorch】Pytorch或者CUDA版本不符合问题解决

  1. 查看CUDA版本。
  2. 查看GPU驱动程序版本。

在这里插入图片描述
例如,我的CUDA版本是10.1。
找到第一行的Driver Version,例如我的是535.129.03 ,对应的GPU驱动程序版本。

  1. 检查二者是否兼容。
    驱动程序与CUDA的兼容性是非常重要的,因为PyTorch和其他基于CUDA的库需要与GPU驱动程序进行交互才能正确运行。如果驱动程序与CUDA不兼容,可能会导致无法正常使用CUDA功能或出现错误。

CUDA版本与GPU驱动的兼容性表格(Linux请参照第一二列,Windows请参照第一三列)如下,请根据自己的版本自行比对:

在这里插入图片描述
在这里插入图片描述

  1. Potorch版本检查命令:
import torchx = torch.rand(5, 3)
print(x)torch.cuda.is_available()
print("cuda: ",torch.cuda.is_available())print(torch.__version__)

在这里插入图片描述
在这里插入图片描述
原因找到了 pytorch版本与cuda版本不对应导致:

pip3 install torch torchvision torchaudio

重新安装pytorch,后可以使用!

在这里插入图片描述

http://www.lryc.cn/news/271226.html

相关文章:

  • 『精』CSS 小技巧之BEM规范
  • vue3-12
  • 操作系统期末复习
  • element el-table实现可进行横向拖拽滚动
  • 【兔子王赠书第14期】《YOLO目标检测》涵盖众多目标检测框架,附赠源代码和全书彩图!
  • WPF 基础入门(样式)
  • Java ArrayList在遍历时删除元素
  • 多模态大模型的前世今生
  • Android studio 花式按键
  • 使用spring boot实现异常的统一返回
  • 2023-12-11 LeetCode每日一题(最小体力消耗路径)
  • PID为1的僵尸进程的产生及清理
  • 043、循环神经网络
  • node使用nodemonjs自动启动项目
  • Ts自封装WebSocket心跳重连
  • 【unity学习笔记】捏人+眨眼效果+口型效果
  • 动态规划 | 最长公共子序列问题
  • RuntimeError: The NVIDIA driver on your system is too old.
  • Java开发过程中的幂等性问题
  • 基于Docker的软件环境部署脚本,持续更新~
  • C#上位机与欧姆龙PLC的通信08----开发自己的通讯库读写数据
  • 【Redis技术专区】「原理分析」探讨Redis6.0为何需要启用多线程
  • simulink代码生成(六)——多级中断的配置
  • 【Minikube Prometheus】基于Prometheus Grafana监控由Minikube创建的K8S集群
  • 无需翻墙|Stable Diffusion WebUI 安装|AI绘画
  • 在FC中手工创建虚拟机模板
  • OpenSSL provider
  • pandas处理双周数据
  • 2023结婚成家,2024借势起飞
  • linux SHELL语句