mirror of
https://github.com/clearml/clearml-agent
synced 2025-01-31 09:06:52 +00:00
14 lines
338 B
Python
14 lines
338 B
Python
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_addoption(parser):
|
||
|
|
parser.addoption('--cpu', action='store_true')
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_configure(config):
|
||
|
|
if not config.option.cpu:
|
||
|
|
return
|
||
|
|
os.environ['PATH'] = ':'.join(p for p in os.environ['PATH'].split(':') if 'cuda' not in p)
|
||
|
|
os.environ['CUDA_VERSION'] = ''
|
||
|
|
os.environ['CUDNN_VERSION'] = ''
|