Skip to content

Commit 047a0ad

Browse files
committed
Remove local GPU Windows hack to ensure script is pristine for PR
1 parent 6151767 commit 047a0ad

1 file changed

Lines changed: 0 additions & 20 deletions

File tree

evaluate_baselines.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -7,26 +7,6 @@
77

88
try:
99
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
10-
import os
11-
import site
12-
import sys
13-
14-
# Inject pip-installed NVIDIA DLLs into Windows PATH so TF 2.10 can use the GPU without a system CUDA install
15-
try:
16-
site_packages = (
17-
site.getsitepackages()[1]
18-
if len(site.getsitepackages()) > 1
19-
else site.getsitepackages()[0]
20-
)
21-
nvidia_base = os.path.join(site_packages, 'nvidia')
22-
dll_paths = [
23-
os.path.join(nvidia_base, 'cudnn', 'bin'),
24-
os.path.join(nvidia_base, 'cublas', 'bin'),
25-
os.path.join(nvidia_base, 'cuda_runtime', 'bin'),
26-
]
27-
os.environ['PATH'] = os.pathsep.join(dll_paths) + os.pathsep + os.environ.get('PATH', '')
28-
except Exception:
29-
pass
3010

3111
import tensorflow as tf
3212

0 commit comments

Comments
 (0)