cuda和pytorch版本不一致

问题：conda install -c pytorch pytorch=1.5 torchvision=0.6.1 cudatoolkit=10.2安装pytorch后运行代码报错step1：查看cuda版本：nvidia-smi显示cuda版本为10.0step2：由于不太想升级cuda版本，只能将torch版本降低，conda install -c pytorch pytorch=1.4.0

江南汪

2673人浏览 · 2021-05-08 12:24:24

江南汪 · 2021-05-08 12:24:24 发布

问题：
conda install -c pytorch pytorch=1.5 torchvision=0.6.1 cudatoolkit=10.2
安装pytorch后运行代码报错

step1：查看cuda版本：nvidia-smi
在这里插入图片描述
显示cuda版本为10.0
step2：由于不太想升级cuda版本，只能将torch版本降低，
conda install -c pytorch pytorch=1.4.0 torchvision=0.5.0 cudatoolkit=10.0

FlagOS智算系统软件栈

欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

不可不知小技巧｜CPP Wrapper 完全指南：让你的 Triton 算子性能再提升一步

通过本文的探讨，我们深入了解了在处理计算量很小的算子时，如何通过 C++ wrapper 来降低 Wrapper+JIT runtime 的开销。通过 C++ wrapper，我们可以直接在 C++ 层面进行类型管理和内存分配，从而避免 Python 调用带来的额外开销。实验数据也证明了这种方法的有效性，性能提升显著。对于大模型推理、端侧部署、高频小算子调用等场景，C++ Wrapper 能让 T

FlagOS智算系统软件栈

10芯齐发：众智FlagOS完成DeepSeek-V4多芯适配，清微智能与曦望实现284B模型版本的适配开源

在FlagOS的统一算子库FlagGems、统一编译器FlagTree及基于FlagScale的多芯片适配支持下，海光、沐曦、华为、摩尔线程（FP8）、昆仑芯、平头哥、天数、英伟达（FP8）、清微、曦望等10款芯片，已经完成 DeepSeek-V4系列模型的跨芯适配及验证。同时，基于 FlagRelease 直接提供了多芯片版本的 DeepSeek-V4-FlagOS 模型版本，标准化 Docke

FlagOS智算系统软件栈

大模型新拐点：FlagOS+Engram 开启算存协同新时代

本文首先介绍 Engram 的核心思想，然后阐述基于 FlagOS 系统软件栈中的训练插件完成的 Engram 架构全链路复现。在此基础上，重点展示 FlagOS 针对 Engram 进行的三大工程优化。实验量化结果表明，FlagOS 对 Engram 的优化在保持额外负载为零的前提下，使吞吐最高提升近 150%。