drewThomasson
commited on
Commit
•
fa6a207
1
Parent(s):
be5dec7
removed sudo from cuda install command
Browse files
app.py
CHANGED
@@ -35,10 +35,10 @@ def install_cuda_if_gpu_detected():
|
|
35 |
print("GPU detected! Proceeding with installation...")
|
36 |
try:
|
37 |
# Update package list
|
38 |
-
subprocess.run([
|
39 |
|
40 |
# Install CUDA dependencies
|
41 |
-
subprocess.run([
|
42 |
|
43 |
print("Installation complete.")
|
44 |
except subprocess.CalledProcessError as e:
|
|
|
35 |
print("GPU detected! Proceeding with installation...")
|
36 |
try:
|
37 |
# Update package list
|
38 |
+
subprocess.run([apt-get', 'update'], check=True)
|
39 |
|
40 |
# Install CUDA dependencies
|
41 |
+
subprocess.run([apt-get', 'install', '-y', 'libcudnn8', 'libcudnn8-dev'], check=True)
|
42 |
|
43 |
print("Installation complete.")
|
44 |
except subprocess.CalledProcessError as e:
|