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

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐