Home > Archives > 2021年12月11日

2021年12月11日

WSL2 で CUDA 使ってみた

Win11 に上げたので WSL2 から CUDA が使えるだろうと,https://docs.nvidia.com/cuda/wsl-user-guide/index.html を参考に WSL2 に CUDA を入れてみた.書かれているコマンドをそのままバージョンだけ 11.5 に変えて:

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.5.0/local_installers/cuda-repo-wsl-ubuntu-11-5-local_11.5.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-5-local_11.5.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-5-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

そしてサンプルをビルド&ラン(書き込み権限がないので手元にコピーして):

cp -r /usr/local/cuda-11.5/samples/4_Finance ./
cp -r /usr/local/cuda-11.5/samples/common ./
cd 4_Finance/BlackScholes
make BlackScholes
../../bin/x86_64/linux/release/BlackScholes

ちゃんと動きましたとさ.めでたし.まあ,サンプルを動かさずとも nvidia-smi を実行して GPU が見えてれば問題ない気もする.

Home > Archives > 2021年12月11日

Search
Feeds

Page Top