diff --git a/python/setup.py b/python/setup.py
index 6a04a4e42..af1fa3068 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -27,10 +27,11 @@ def get_llvm():
         return '', ''
     # download if nothing is installed
     name = 'clang+llvm-11.0.1-x86_64-linux-gnu-ubuntu-16.04'
-    dir = '/tmp'
+    dir = os.path.join(os.environ["HOME"], ".triton", "llvm")
     llvm_include_dir = '{dir}/{name}/include'.format(dir=dir, name=name)
     llvm_library_dir = '{dir}/{name}/lib'.format(dir=dir, name=name)
     if not os.path.exists(llvm_library_dir):
+        os.makedirs(dir, exist_ok=True)
         try:
             shutil.rmtree(os.path.join(dir, name))
         except Exception:
diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index 60feb1740..90a031a30 100644
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -1158,8 +1158,7 @@ class DependenciesFinder(ast.NodeVisitor):
 
 
 def default_cache_dir():
-    import getpass
-    return f'/tmp/triton_{getpass.getuser()}'
+    return os.path.join(os.environ["HOME"], ".triton", "cache")
 
 
 class JITFunction: