diff --git a/install/mindspore_gpu_install_conda.md b/install/mindspore_gpu_install_conda.md index 8725e4c956acea336ed80b537f94d641ac1413d3..0f08d90a2e92c18fd4c9150302b6eae462ccff42 100644 --- a/install/mindspore_gpu_install_conda.md +++ b/install/mindspore_gpu_install_conda.md @@ -174,6 +174,7 @@ conda install mindspore -c mindspore -c conda-forge ```bash 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路径。 diff --git a/install/mindspore_gpu_install_conda_en.md b/install/mindspore_gpu_install_conda_en.md index b0cf741801c143ea217db862e2f3bff7a15a2778..fb0ae2d542b52dc59bbce01308a1484e4f475e36 100644 --- a/install/mindspore_gpu_install_conda_en.md +++ b/install/mindspore_gpu_install_conda_en.md @@ -174,6 +174,7 @@ Before running MindSpore GPU version, please make sure that installation path of ```bash 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 ``` If a different version of CUDA have been installed or the CUDA installation path is different, replace `/usr/local/cuda-11.6` in the above command with the currently installed CUDA path. diff --git a/install/mindspore_gpu_install_nightly.md b/install/mindspore_gpu_install_nightly.md index 3d87f7c07fb883d05b641bee079eebcf3c88ebf9..53852cbc88f109c900dd486dfa0b4774118efb72 100644 --- a/install/mindspore_gpu_install_nightly.md +++ b/install/mindspore_gpu_install_nightly.md @@ -190,6 +190,7 @@ pip install mindspore-dev -i https://pypi.tuna.tsinghua.edu.cn/simple ```bash 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路径。 diff --git a/install/mindspore_gpu_install_nightly_en.md b/install/mindspore_gpu_install_nightly_en.md index 7b580eb49abd38ac09639efbe2d1284bde514848..951274a617957ba317ab4e880e81220eddd492bb 100644 --- a/install/mindspore_gpu_install_nightly_en.md +++ b/install/mindspore_gpu_install_nightly_en.md @@ -190,6 +190,7 @@ Before running MindSpore GPU version, please make sure that installation path of ```bash 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 ``` If a different version of CUDA have been installed or the CUDA installation path is different, replace `/usr/local/cuda-11.6` in the above command with the currently installed CUDA path. diff --git a/install/mindspore_gpu_install_pip.md b/install/mindspore_gpu_install_pip.md index 268121e23b3316ce4bcc1b7917dd23fb5c83d7d7..91e9c209a9d39246c9ea29ef98f10cb16783636c 100644 --- a/install/mindspore_gpu_install_pip.md +++ b/install/mindspore_gpu_install_pip.md @@ -195,6 +195,7 @@ pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/${MS_VERSION}/Mi ```bash 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路径。 diff --git a/install/mindspore_gpu_install_pip_en.md b/install/mindspore_gpu_install_pip_en.md index b6d2cbe22a58a9edb72283031902dba3d077d3fc..fdf0685dfda1cb4c863adf6527238311e8cb5eac 100644 --- a/install/mindspore_gpu_install_pip_en.md +++ b/install/mindspore_gpu_install_pip_en.md @@ -195,6 +195,7 @@ Before running MindSpore GPU version, please make sure that installation path of ```bash 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 ``` If a different version of CUDA have been installed or the CUDA installation path is different, replace `/usr/local/cuda-11.6` in the above command with the currently installed CUDA path.