https://www.mindspore.cn/install#command

cuda版本要求为11.6,之前服务器已经安装了cuda12.4,所以需要再安装一个cuda11.6版本

运行MindSpore GPU版本前,请确保nvcc的安装路径已经添加到PATHLD_LIBRARY_PATH环境变量中,如果没有添加,以安装在默认路径的CUDA11为例,可以执行如下操作:

export PATH=/usr/local/cuda-11.6/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.6

如果之前安装了其他CUDA版本,或者CUDA安装路径不同,只需替换以上命令中的/usr/local/cuda-11.6为当前安装的CUDA路径。

实际安装测试结果

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐