wsl安裝gpu版mindspore（二）

沒辦法，網上的一鍵安裝不适用于wsl

我們自己來

參考

https://blog.csdn.net/qq_20291997/article/details/106897324

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo apt install build-essential
sudo sh cuda_11.1.1_455.32.00_linux.run
           

安裝cuda

安裝cuda時千萬記住不要安裝驅動，不要安裝驅動，不要安裝驅動！！！！其餘選項該怎麼裝怎麼裝，就是不要裝驅動！！！！Driver相關一定要N或者X掉！！！！！！

但是貌似這個版本就沒有Driver選項

安裝後一定有提示，Driver: Not Selected。

echo -e "export PATH=/usr/local/cuda-11.1/bin:\$PATH" >> ~/.bashrc
echo -e "export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:\$LD_LIBRARY_PATH" >> ~/.bashrc
source ~/.bashrc
           

在這裡找配套cudnn的版本

https://developer.download.nvidia.com/compute/cudnn/secure/8.5.0/local_installers/11.7/cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz

下載下傳了這個tar檔案後，我們

xz -d cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz

tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.1/include

sudo cp cuda/lib/libcudnn* /usr/local/cuda-11.1/lib64

sudo chmod a+r /usr/local/cuda-11.1/include/cudnn.h /usr/local/cuda-11.1/lib64/libcudnn*

通路 https://docs.conda.io/en/latest/miniconda.html

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sudo bash Miniconda3-latest-Linux-x86_64.sh
/usr/local/miniconda3
/usr/local/miniconda3/bin/conda init bash
           

conda安裝完成

#conda create -n mindspore_py37 python=3.7.5 -y
#conda activate mindspore_py37
可以就安裝在base環境
           

python自動是3.9.12

安裝mpi，便于遠端

curl -O https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.3.tar.gz
tar xzf openmpi-4.0.3.tar.gz
cd openmpi-4.0.3
./configure --prefix=/usr/local/openmpi-4.0.3
make
sudo make install
echo -e "export PATH=/usr/local/openmpi-4.0.3/bin:\$PATH" >> ~/.bashrc
echo -e "export LD_LIBRARY_PATH=/usr/local/openmpi-4.0.3/lib:\$LD_LIBRARY_PATH" >> ~/.bashrc
source ~/.bashrc
cd -
echo 'export PATH=/home/kewei/.local/bin:$PATH' >> ~/.bashrc
python -c "import mindspore;mindspore.run_check()"

           

第一次執行驗證語句需要幾分鐘，請耐心等待。

成功！