windows10 + wsl2,使用NVIDIA gpu

windows10 + wsl2,使用NVIDIA gpu

安装wsl2

参考 https://www.youtube.com/watch?v=_fntjriRe48

依赖：需要win10系统满足一定的版本号。

在Windows开始里分别输入如下：

• winver，检查Windows 10版本，确保Windows10版本满足要求（Version 1903 or higher, with Build 18362 or higher即可）
• windows feature，打开“启用或关闭Windows功能”，勾选上“虚拟机平台”和“适用于Linux的Windows子系统”

然后参照 https://docs.microsoft.com/en-us/windows/wsl/install-win10 分别安装wsl并使能wsl2、安装wsl_update包、安装Ubuntu等。

安装NVIDIA驱动，并在wsl2里设置CUDA toolkit

参考 https://docs.microsoft.com/en-us/windows/win32/direct3d12/gpu-cuda-in-wsl
参考 https://docs.nvidia.com/cuda/wsl-user-guide/index.html
参考 https://www.youtube.com/watch?v=PdxXlZJiuxA

依赖：Windows10 insider preview版本并切换到Dev channel

说明1：截止到写当前文章之时，要在wsl2里使用NVIDIA cuda，需要加入windows10 insider preview并切换到Dev channel.（否则会出现wsl2里load libcuda.so.1库失败等情况）

说明2：若使用nvidia docker，参考上面链接按照官方的步骤安装。

下面是不使用nvidia docker的方式：

• 切换到windows10 insider preview版本Dev channel，更新系统
• Windows上，从 https://developer.nvidia.com/cuda/wsl 下载NVIDIA驱动并安装
• 在wsl2里，设置CUDA Toolkit，命令如下:

# 下面命令摘自NVIDIA官网

$ apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

$ sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

# 如果网络不顺畅的话，在这里可以把cuda.list里所配置的nvidia的repo地址换成阿里镜像 https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64

$ apt-get update

$ apt-get install -y cuda-toolkit-11-0  (安装自己需要的cuda toolkit版本，例如10-0版本)

额外地，如果安装有多个cuda toolkit版本的话，可以通过手动改变软连接/usr/local/cuda来指向当前想要使用的cuda版本。例如：

$ cd /usr/local/
$ ls -l
lrwxrwxrwx  cuda -> cuda-10.0/
drwxr-xr-x  cuda-10.0/
drwxr-xr-x  cuda-10.1/
......

$ sudo rm -fv cuda
$ sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

Ubuntu(wsl2) 安装cuDNN

以cuDNN 10.0为例：

从NVIDIA cuDNN官网 或去cuDNN档案里下载对应版本的文件。

tar xaf ~/Download/cudnn-10.0-linux-x64-v7.6.5.32.tgz
sudo cp -Pv include/cudnn.h /usr/local/cuda-10.0/include/
sudo cp -Pv lib64/* /usr/local/cuda-10.0/lib64/
# 如果拷贝后的文件没有读权限，加上chmod a+r ...

完成上述拷贝cuDNN库后，记得执行一下ldconfig：

sudo ldconfig

在wsl2里使用Windows上的vscode

参考 https://docs.microsoft.com/en-us/windows/wsl/tutorials/wsl-vscode
参考 https://www.youtube.com/watch?v=A0eqZujVfYU

如果想要使用vscode编辑wsl2里的代码：

• 在Windows上安装vscode，注意安装的时候选择Add to PATH选项
• 安装完成后，在vscode里安装extension: Remote Development
• 直接在wsl2的linux里运行code .，会打开vscode，并自动安装相关扩展

pycharm连接使用wsl2

参考 https://www.jetbrains.com/help/pycharm/using-wsl-as-a-remote-interpreter.html

在"Settings - Project <project name> | Python Interpreter."里Add选择wsl里的python。

---

------更新于2020.10.27