MannLabs / alphapept

A modular, python-based framework for mass spectrometry. Powered by nbdev.

Home Page:https://mannlabs.github.io/alphapept/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA initialization error

im281 opened this issue · comments

commented

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()`