tforgione
/
nushell
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix deep learning GPU

This commit is contained in:
Thomas Forgione 2024-02-15 13:31:20 +01:00
parent d8559197c7
commit c3303a9842
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
env.nu
View File

@ -252,9 +252,6 @@ $env.RUST_BACKTRACE = full
$env.GCLONE_PATH = $env.HOME + '/git' $env.GCLONE_PATH = $env.HOME + '/git'
$env.GCLONE_FORCE_SSH = true $env.GCLONE_FORCE_SSH = true
# Python virtual env
$env.VIRTUAL_ENV = $env.HOME + '/.venv'
# Don't put ugly go directory in my home # Don't put ugly go directory in my home
$env.GOPATH = $env.HOME + '/.go' $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 '/some/path')
$env.PATH = ($env.PATH $env.PATH = ($env.PATH
| split row (char esep) | split row (char esep)
| prepend ('/opt/cuda/bin')
| prepend ($env.HOME + '/.config/polymny/bin') | prepend ($env.HOME + '/.config/polymny/bin')
| prepend ($env.HOME + '/.config/dotfiles/bin') | prepend ($env.HOME + '/.config/dotfiles/bin')
| prepend ($env.HOME + '/.config/dotfiles/bin-extra') | prepend ($env.HOME + '/.config/dotfiles/bin-extra')
@ -270,6 +268,13 @@ $env.PATH = ($env.PATH
| prepend ($env.HOME + '/.venv/bin') | 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") { $env.XDG_SESSION_TYPE = if (tty | str starts-with "/dev/pts") {
"x11" "x11"
} else { } else {