You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+13Lines changed: 13 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -40,6 +40,19 @@ Follow these steps in case you'd like to clone and run the DALL-E playground loc
40
40
it `npm start`
41
41
7. Copy backend's url from step 5 and paste it in the backend's url input within the web app
42
42
43
+
## Local development/use with Windows WSL2
44
+
45
+
Window's WSL2 Linux layer has some unique issues getting running with GPU support. Nvidia CUDA drivers are installed on the Windows side instead of Linux, but jax does not see the GPU without compiling from source. Here are extra instructions to get jax compiled.
46
+
47
+
1. Have a recent NVIDIA GeForce Game Ready or NVIDIA RTX Quadro driver installed in Windows
48
+
2. In Linux: Install Nvidia's CUDA toolkit, [WSL instructions](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#wsl-installation)
49
+
3. In Linux: Install Nvidia's CuDNN library: [instructions](https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html)
50
+
4. In Linux: Build and install both `jaxlib` and `jax` from source, remember to enable cuda during compilation with `python3 build/build.py --enable_cuda`[instructions](https://jax.readthedocs.io/en/latest/developer.html)
51
+
5. In compiling `jaxlib`, you might hit a broken configuration file, solution here: https://github.com/google/jax/issues/11068
52
+
6. Follow local development instructions above
53
+
54
+
WSL2 installs are fairly bare bones, expect to install packages like `npm`, `python3-pip` and many others to get things working
55
+
43
56
## Local development with Docker-compose
44
57
45
58
1. Make sure you have [docker](https://docs.docker.com/get-docker/) and [The NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html) installed
0 commit comments