akhaliq HF Staff commited on
Commit
cfa1ad3
·
1 Parent(s): 5f1cc9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,10 +2,10 @@ import os
2
  import os
3
 
4
  os.system("wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
5
- os.system("dpkg-i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb --user")
6
- os.system("apt-key add --user /var/cuda-repo-9-2-local/7fa2af80.pub")
7
- os.system("apt-get update --user")
8
- os.system("apt-get install cuda --user")
9
 
10
  os.system("nvcc --version")
11
  print(os.environ.get('CUDA_PATH'))
 
2
  import os
3
 
4
  os.system("wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
5
+ os.system("sudo dpkg-i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
6
+ os.system("sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub")
7
+ os.system("sudo apt-get update")
8
+ os.system("sudo apt-get install cuda")
9
 
10
  os.system("nvcc --version")
11
  print(os.environ.get('CUDA_PATH'))