# jaxlibWithCuda not using CUDA

**URL:** https://discourse.nixos.org/t/jaxlibwithcuda-not-using-cuda/36873
**Category:** Help
**Created:** [December 14, 2023, 4:35pm UTC](https://discourse.nixos.org/t/jaxlibwithcuda-not-using-cuda/36873 "2023-12-14T16:35:26Z")
**Posts on this page:** 1
**Showing post:** 2

<div class="post-metadata">

### Author: ![euclaise](https://discourse.nixos.org/user_avatar/discourse.nixos.org/euclaise/32/12209_2.png) [@euclaise](https://discourse.nixos.org/u/euclaise)
#### Post date: [December 14, 2023, 4:59pm UTC](https://discourse.nixos.org/t/jaxlibwithcuda-not-using-cuda/36873/2 "2023-12-14T16:59:55Z")

</div>

Update:

Replacing the first line with `{ pkgs ? import <nixpkgs> { config.allowUnfree = true; config.cudaSupport = true } }:` causes it to perform a lot of tests.

However, the same error persists, although now wandb must be removed since it fails some tests.

---

_[View the full topic](https://discourse.nixos.org/t/jaxlibwithcuda-not-using-cuda/36873)._