Friday, January 21, 2022

WSL with CUDA support

Windows Subsystem for Linux with CUDA support + TensorFlow

Let me save you many hours of pain and just give you the rundown.

* You MUST be on Windows 10 21H2. 21H2 cannot currently be updated to (yeah, it's because they are pushing Windows 11) so you need to download it as an ISO + install with USB.

* Install the latest "game ready driver" for your card (you don't need a special driver despite the nvidia documentation, I got mine here: https://ift.tt/1OtSyri).

* In general you should ignore the nvidia documentation as they make it almost impossible to figure out what combination of cuda & cudnn versions you need.

Steps to install:

1. Open CMD as administrator.

2. Run: wsl --install -d ubuntu

3. Open the Ubuntu WSL and enter the following commands:

# Install CUDA

sudo apt-key adv --fetch-keys https://ift.tt/3tT8BtV

sudo sh -c 'echo "deb https://ift.tt/3fKz3Of /" > /etc/apt/sources.list.d/cuda.list'

sudo apt-get update

sudo apt-get --yes install cuda-toolkit-11-2 cuda-toolkit-11-2

sudo sh -c 'echo "deb https://ift.tt/3rDFS9F /" > /etc/apt/sources.list.d/nvidia-machine-learning.list'

sudo apt-get update

sudo apt install cuda-11-2 libcudnn8 libcudnn8-dev

sudo apt install libnvinfer8 libnvinfer-dev libnvinfer-plugin8

# Verify CUDA installation

cd /usr/local/cuda-11.2/samples/4_Finance/BlackScholes

sudo make

./BlackScholes

# Install TensorFlow pip3 install tensorflow-gpu

If you are reading this in the future I have some advice for you:

* Look at this page: https://ift.tt/35f68Qj

* Use apt-cache search <THING> to see the possible versions, allowing you to guess what you need to install. Good luck.



from Hacker News https://ift.tt/3GRSYGZ

No comments:

Post a Comment

Note: Only a member of this blog may post a comment.