Windows Subsystem for Linux (Unbunt 23.04) 安装 Cuda 驱动

@[TOC](Windows Subsystem for Linux (Unbunt 23.04) 安装 Cuda 驱动)

安装 CUDA Toolkit

refer: https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl

refer: https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=runfile_local

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run

设置环境变量

sudo vi ~/.bashrc

--- add
export CUDA_HOME=/usr/local/cuda-12.1
export CuDNN_HOME=/usr/local/cuda-12.1/include
export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH
---

source ~/.bashrc

验证安装是否成功

watch -n 1 nvidia-smi

通过示例程序验证安装是否成功

创建虚拟环境，

python3 -m venv venv
source ./venv/bin/activate

安装依赖，

pip3 install numba

创建验证程序，

cat << EOF > verifycuda.py
from numba import cuda
print(cuda.gpus)
EOF

执行验证程序，

python3 verifycuda.py

如果输出以下结果，说明安装成功，

<Managed Device 0>

完结！