conda环境安装mindspore

检查各个库版本

CUDA:nvcc -V (11.5)
GCC:gcc --version (12.3)
python:python -V (3.8.19)

conda新环境

安装cuda

mindspore目前支持的 CUDA 版本只有 10.1,11.1 和 11.6 三个

查看都有哪些版本
conda search cudatoolkit
conda install cudatoolkit==11.1.1(指定版本)

cudnn安装

官网查询与cuda的对对应版本关系:https://developer.nvidia.com/rdp/cudnn-archive

查看都有哪些版本
conda search cudnn
conda install cudnn==8.9.7(指定的版本)

下载失败后使用国内临时源

conda install cudnn=8.9.7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

安装MindSpore

确认您处于Conda虚拟环境中,并执行如下命令安装最新版本的MindSpore。如需安装其他版本,可参考版本列表在mindspore=后指定版本号

conda install mindspore -c mindspore -c conda-forge

查询官网找到cuda对应的版本:https://www.mindspore.cn/install/#command

conda install mindspore=2.2.12 -c mindspore -c conda-forge

验证是否成功安装

运行MindSpore GPU版本前,请确保nvcc的安装路径已经添加到PATH与LD_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

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

python -c "import mindspore;mindspore.set_context(device_target='GPU');mindspore.run_check()"

如果输出:

MindSpore version: 版本号
The result of multiplication calculation is correct, MindSpore has been installed on platform [GPU] successfully!

posted @ 2024-03-26 23:57  妙妙屋~  阅读(121)  评论(0)    收藏  举报