Устанавливаем cudа на WSL2 по инструкции.
- Проверяем, что установлены поcледние драйверы Nvidia, если нужно обновить, то ставить с официального сайта https://www.nvidia.com/Download/index.aspx
- Проверяем последнюю версию wsl
wsl.exe --update
- Ставим cuda
sudo apt-key del 7fa2af80
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/12.0.0/local_installers/cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
- Смотрим видео с youtube
Краткая инструкция по видео
- Ставим miniconda3 с официального сайта https://docs.conda.io/en/latest/miniconda.html#linux-installers
wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.1.0-1-Linux-x86_64.sh
bash Miniconda3-py310_23.1.0-1-Linux-x86_64.sh
- С сайта https://rapids.ai/start.html#get-rapids копируем команду для установки rapids
conda create -n rapids-23.02 -c rapidsai -c conda-forge -c nvidia \
rapids=23.02 python=3.10 cudatoolkit=11.8
- Процесс установки занимает некоторое время
- Выполняем команду conda activate rapids-23.02
- Запускаем jupyter-notebook
Если jupyter notebook не установлено, то выполняем:
sudo apt install python3-pip python3-dev
pip install jupyter