GPU memory is used while running inference.py changing the device to cpu

Question

GPU memory is used while running inference.py changing the device to cpu

Opened this issue 6 months ago · 1 comments

srilekha1993 commented 6 months ago

Hi,
I observed while running the command [ time python -m inference --config default_inference_args.yaml --protein_ligand_csv data/test_1.csv --out_dir results_L_cpu/user_predictions_small] the GPU memory is used while the device is changed to 'cpu' in inference.py.
Can anyone help me out how to completely run the above script in cpu only?

Answer 1 · 2024-04-26T15:01:02.000Z

The simplest thing would be to hide by environment variable

export CUDA_VISIBLE_DEVICES=-1
time python -m inference --config default_inference_args.yaml --protein_ligand_csv data/test_1.csv --out_dir results_L_cpu/user_predictions_small