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

[Anonymous]

Я хочу использовать свой графический процессор, отрегулировав количество рабочих процессов, но у меня проблема с количеством рабочих процессов > 0.
test_loader = DataLoader(test_dataset,atch_size=32, shuffle =False, num_workers=0) — нет проблем
test_loader = DataLoader(test_dataset, пакетный_размер=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