Sergidev commited on
Commit
99b094e
1 Parent(s): 621dcee

Update usr.

Browse files
Files changed (1) hide show
  1. start.py +1 -1
start.py CHANGED
@@ -3,5 +3,5 @@ import subprocess
3
  # commented because the existing llama-cpp-python package was renoved fron requirements.txt
4
  subprocess.run("pip uninstall -y llama-cpp-python", shell=True)
5
 
6
- install_command = "CMAKE_ARGS='-DLLAMA_CUDA=on -DCMAKE_CUDA_COMPILER=/home/user/local/cuda/bin/nvcc' pip install llama-cpp-python"
7
  subprocess.run(install_command, shell=True)
 
3
  # commented because the existing llama-cpp-python package was renoved fron requirements.txt
4
  subprocess.run("pip uninstall -y llama-cpp-python", shell=True)
5
 
6
+ install_command = "CMAKE_ARGS='-DLLAMA_CUDA=on -DCMAKE_CUDA_COMPILER=./usr/local/cuda/bin/nvcc' pip install llama-cpp-python"
7
  subprocess.run(install_command, shell=True)