Skip to content
Snippets Groups Projects
Commit ddde4efe authored by Jan Ebert's avatar Jan Ebert
Browse files

Add PyTorch kernel cache path

parent 110c0a66
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 1 addition and 0 deletions
README.md
+ 1
− 0
View file @ ddde4efe
...@@ -244,6 +244,7 @@ directories mentioned here. ...@@ -244,6 +244,7 @@ directories mentioned here.
Here are some of the variables concerning various libraries and their Here are some of the variables concerning various libraries and their
default values. default values.
- PyTorch kernels: `PYTORCH_KERNEL_CACHE_PATH="$HOME"/.cache/torch/kernels`
- PyTorch Hub: `TORCH_HOME="$HOME"/.cache/torch/hub` - PyTorch Hub: `TORCH_HOME="$HOME"/.cache/torch/hub`
- PyTorch extensions: `TORCH_EXTENSIONS_DIR="$HOME"/.cache/torch_extensions` - PyTorch extensions: `TORCH_EXTENSIONS_DIR="$HOME"/.cache/torch_extensions`
- Triton (PyTorch dependency): `TRITON_CACHE_DIR="$HOME"/.triton/cache` - Triton (PyTorch dependency): `TRITON_CACHE_DIR="$HOME"/.triton/cache`
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment