WSL2中使用GPU

雨点打透心脏的1/2处 2022-11-12 04:22 621阅读 0赞

### 在WSL2上安装CUDA和NVIDIA HPC SDK ###

*  1.WSL2和Ubuntu的安装
 *  2.安装显卡驱动
 *  3.在WSL2中安装CUDA
 *  4.安装 NVIDIA HPC SDK

Windows10内部预览版20145及之后的版本的WSL2支持GPU加速，NVIDIA HPC SDK是英伟达公司推出的免费版的编译器，该编译器支持CUDA和OpenACC，同时支持C/C++和Fortran。下面是在WSL2上安装CUDA和NVIDIA HPC SDK的步骤。

# 1.WSL2和Ubuntu的安装 #

根据https://docs.microsoft.com/zh-cn/windows/wsl/install-win10上的教程，完成WSL2的安装，并设置WSL2为默认，然后在Micsoft store中搜索Ubuntu进行安装。

# 2.安装显卡驱动 #

在win10系统下安装NVIDIA显卡驱动，在https://developer.nvidia.com/cuda/wsl页面选择对应的显卡驱动。在win10下安装完显卡驱动后，不需要在WSL2系统中安装驱动。安装完驱动后在WSL2中使用电脑的GPU了，随后在WSL2中安装CIDA就可以进行编程了。

# 3.在WSL2中安装CUDA #

安装CUDA之前需要先安装GCC。

sudo apt-get install gcc

安装以下步骤安装CUDA，由于第2步已经安装了显卡驱动，因此此处安装CUDA时，只需要安装cuda-toolkit，不需要完整安装。

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
    sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
    wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-wsl-ubuntu-11-1-local_11.1.1-1_amd64.deb
    sudo dpkg -i cuda-repo-wsl-ubuntu-11-1-local_11.1.1-1_amd64.deb
    sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-1-local/7fa2af80.pub
    sudo apt-get update
    sudo apt-get install -y cuda-toolkit-11-1

安装完成之后配置环境变量，CUDA默认安装目录为/usr/local，根据自己情况改变路径。

export PATH=/usr/local/cuda-11.1/bin${
        PATH:+:${
        PATH}}
    CUDA_HOME=/usr/local/cuda-11.1; export CUDA_HOME
    export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${
        LD_LIBRARY_PATH:+:${
        LD_LIBRARY_PATH}}

# 4.安装 NVIDIA HPC SDK #

安装CUDA之前需要先安装g++。

sudo apt-get install g++

在NVIDIA官网https://developer.nvidia.com/hpc-sdk下载NVIDIA HPC SDK编译器，该版本为社区版，可以免费使用。下载完成之后，将安装包解压，在解压后的目录直接安装：

sudo ./install

安装完成之后配置环境变量，HPC SDK的默认安装目录为/opt/nvidia，根据自己情况改变路径。

NVARCH=`uname -s`_`uname -m`; export NVARCH
    NVCOMPILERS=/opt/nvidia/hpc_sdk; export NVCOMPILERS
    MANPATH=$MANPATH:$NVCOMPILERS/$NVARCH/20.11/compilers/man; export MANPATH
    PATH=$NVCOMPILERS/$NVARCH/20.11/compilers/bin:$PATH; export PATH

至此NVIDIA HPC SDK在WSL2上安装成功。