fastai1
fastai1 copied to clipboard
Exception of pin_memory
I'm having a constant exception after running the following code part:
from fastai.vision import * import warnings warnings.filterwarnings('ignore') path = 'G:/DataScienceProject/Kaggle-Prostate-cANcer-graDe-Assessment/train' folderList = os.listdir(path) data = ImageDataBunch.from_folder(path, train=".", test="../cv", valid_pct=0.2, classes=folderList)
from fastai.metrics import error_rate # 1 - accuracy learn = create_cnn(data, models.resnet34, metrics=accuracy) defaults.device = torch.device('cuda') learn.fit_one_cycle(10)
Fastai version: 1.0.61
Exception:
RuntimeError: Caught RuntimeError in pin memory thread for device 0.
Original Traceback (most recent call last):
File "C:\Users\User\AppData\Roaming\Python\Python38\site-packages\torch\utils\data_utils\pin_memory.py", line 31, in _pin_memory_loop
data = pin_memory(data)
File "C:\Users\User\AppData\Roaming\Python\Python38\site-packages\torch\utils\data_utils\pin_memory.py", line 55, in pin_memory
return [pin_memory(sample) for sample in data]
File "C:\Users\User\AppData\Roaming\Python\Python38\site-packages\torch\utils\data_utils\pin_memory.py", line 55, in
-Gilad