Используйте GPU для кода Python в VSCODE

conda install numba & conda install cudatoolkit
Colorful Caribou