Итак, когда я запускаю cuda.select_device(0)
, а затем cuda.close()
. Pytorch не может снова получить доступ к GPU, я знаю, что есть способ, чтобы PyTorch мог снова использовать GPU без необходимости перезапуска ядра. Но я забыл как. Кто-нибудь еще знает?