CUDA

主要是說 CUDA 應用程式可以直接在 WSL 上跑然後唯一需要安裝驅動的只有 Windows (NVIDIA GeForce Game Ready Driver) 重點是 WSL 裡面絕對不要再安裝 Driver 大概是說 WSL 裡面就已經會使用 Winidows 的 Driver 了如果再從裡面安裝會覆蓋掉所以裡頭只要安裝 CUDA Toolkit 就好了

WSL 2

用 WSL 安裝 Ubuntu, 這邊不贅述

Ubuntu

CUDA

首先刪除舊的 GPG key:

sudo apt-key del 7FA2AF80

然後下載 CUDA Toolkit

WSL-Ubuntu

有三種安裝方式:

deb (local)：如果需要常常安裝，用 LOCAL 方法可以避免每次重新下載 ~4GB 的檔案
deb (network)：每次安裝都會重新下載 ~4GB 的檔案
runfile (local)：需自行安裝 GCC，否則報錯： Missing gcc. gcc is required to continue

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-5

nvidia-smi

編輯 ~/.bashrc:

nano ~/.bashrc

添加以下內容:

# nvidia-smi
export PATH=$PATH:/usr/lib/wsl/lib/
# nvcc
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

驗證安裝:

source ~/.bashrc
nvidia-smi
nvcc --version

REF

wsl-user-guide

Last modified: 31 May 2024