WatsonTang98
/

baseline_null

Model card Files Files and versions Community

WatsonTang98 commited on Jun 9

Commit

056c4ed

•

1 Parent(s): ae67fe6

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -157,7 +157,7 @@ def download_packages(packages, folder):
 		print(f"Failed to download packages. Error: {e}")
 # Set CUDA environment variables
-os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
 os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
 os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')

 		print(f"Failed to download packages. Error: {e}")
 # Set CUDA environment variables
+os.environ['CUDA_HOME'] = '/usr/local/cuda'
 os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
 os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')