From c3303a9842bda957d1a035d444152441f6d9f832 Mon Sep 17 00:00:00 2001
From: Thomas Forgione <thomas@forgione.fr>
Date: Thu, 15 Feb 2024 13:31:20 +0100
Subject: [PATCH] Fix deep learning GPU

---
 env.nu | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/env.nu b/env.nu
index 8f4d99f..f0d616e 100644
--- a/env.nu
+++ b/env.nu
@@ -252,9 +252,6 @@ $env.RUST_BACKTRACE = full
 $env.GCLONE_PATH = $env.HOME + '/git'
 $env.GCLONE_FORCE_SSH = true
 
-# Python virtual env
-$env.VIRTUAL_ENV = $env.HOME + '/.venv'
-
 # Don't put ugly go directory in my home
 $env.GOPATH = $env.HOME + '/.go'
 
@@ -262,6 +259,7 @@ $env.GOPATH = $env.HOME + '/.go'
 # $env.PATH = ($env.PATH | split row (char esep) | prepend '/some/path')
 $env.PATH = ($env.PATH
     | split row (char esep)
+    | prepend ('/opt/cuda/bin')
     | prepend ($env.HOME + '/.config/polymny/bin')
     | prepend ($env.HOME + '/.config/dotfiles/bin')
     | prepend ($env.HOME + '/.config/dotfiles/bin-extra')
@@ -270,6 +268,13 @@ $env.PATH = ($env.PATH
     | prepend ($env.HOME + '/.venv/bin')
 )
 
+# Deep learning on GPU
+$env.XLA_FLAGS = '--xla_gpu_cuda_data_dir=/opt/cuda'
+
+# Python virtual env
+$env.VIRTUAL_ENV = $env.HOME + '/.venv'
+
+
 $env.XDG_SESSION_TYPE = if (tty | str starts-with "/dev/pts") {
     "x11"
 } else {