CUDA initialization error
im281 opened this issue · comments
running multiple files in the test_gpu.py script results in a cuda initialization error. A single file can be run but it is not using GPU.
`
def main():
mode = 'numba-multithread' #'cuda' #sys.argv[1]
print(f"Testing with mode {mode}")
global alphapept
alphapept.performance.set_compilation_mode(mode)
alphapept.performance.set_worker_count(0)
importlib.reload(alphapept.feature_finding)
#delete_folder(tmp_folder)
#create_folder(tmp_folder)
for file in FILE_DICT:
target = os.path.join(tmp_folder, file)
if not os.path.isfile(target):
wget.download(FILE_DICT[file], target)
settings = load_settings(DEFAULT_SETTINGS_PATH)
settings['experiment']['file_paths'] = [os.path.join(tmp_folder, 'test1.raw')]#,
#os.path.join(tmp_folder, 'test2.raw'),
#os.path.join(tmp_folder, 'test3.raw')]
#settings['experiment']['file_paths'] = [os.path.join(tmp_folder, 'thermo_IRT.raw')]
settings['experiment']['fasta_paths'] = [os.path.join(tmp_folder, 'IRT_fasta.fasta')]
import alphapept.interface
#settings = load_settings('/home/ubuntu/apps/alphapept/test_settings.yaml')
settings_ = alphapept.interface.import_raw_data(settings)
start = time()
settings_ = alphapept.interface.feature_finding(settings)
end = time()
te = end-start
print(f'Time elapsed {te}')
if name == "main":
main()`