Open wuwei-bit opened 3 months ago
看起来您在尝试使用OneFlow的 nightly 安装包,nightly 安装包是使用 cuda 11.8 针对 75,80,86 三种 CUDA 架构设备编译,您的设备是61 CUDA架构的设备,且cuda版本为11.5,使用中会有问题。OneFlow 框架是支持 10.0 以上版本 cuda,支持 60 以上CUDA架构 设备编译的,您可以选择从源代码编译 OneFlow 使用,在CMake 命令中通过如 -DCMAKE_CUDA_ARCHITECTURES=61
指定 CUDA 架构,通过如 -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda -DCUDNN_ROOT_DIR=/usr/local/cudnn
指定使用的cuda和cudnn路径。我们之后也会更新README.md中的描述。
Description
oneflow 版本为 0.9.1.dev20240203+cu118
cuda版本为 11.5
显卡信息
运行“快速上手”教程中的代码:https://docs.oneflow.org/master/basics/01_quickstart.html# 报错如下:
oneflow使用 cuda11.8 编译的,按理说可以进行向下兼容 cuda11.5。但是为什么报错信息中说应该使用更新版本的 cuda 重新编译 oneflow? 我应该如何处理这个问题? 另外, README.md 中建议使用 cuda 10.2, 这个是文档更新问题吗?
Alternatives
No response