Ошибка в загрузчике данных PyTorch с num_workers>0 в VSC под WSL ⇐ Python

[Anonymous]

Я хочу использовать свой графический процессор, отрегулировав количество рабочих процессов, но у меня проблема с количеством рабочих процессов > 0.

Код: Выделить всё

test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False, num_workers=0)

 – нет проблем

Код: Выделить всё

test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False, num_workers=1)

— RuntimeError: ошибка CUDA: ошибка инициализации

Набор данных — MNIST. Размер: 50% / 50% обучение/тест
Это NVIDIA GeForce RTX 3060 Ti, Torch 2.0.1 + CUDA 1.18
Память графического процессора 8191,50 МБ
Я работаю в коде Visual Studio в WSL.
Здесь ссылка на аналогичный случай.
Ошибка при использовании pytorch num_workers в colab. когда num_workers > 0
Есть предложения?
Спасибо!
>>>РЕДАКТИРОВАТЬ

Подробнее здесь: https://stackoverflow.com/questions/789 ... -under-wsl