I want to utilize my GPU by adjusting the workers number, but I have a problem with the number of workers > 0.
test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False, num_workers=0)
– no problem
test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False, num_workers=1)
– RuntimeError: CUDA error: initialization error
Data set is MNIST. The size is 50% / 50% train/test
Its NVIDIA GeForce RTX 3060 Ti, Torch 2.0.1 + CUDA 1.18
GPU memory 8191.50 MB
I work within Visual Studio Code in WSL
Here a reference a to similar case.
Error when using pytorch num_workers in colab. when num_workers > 0
Any suggestions?
Thank you!
1