fastai1 icon indicating copy to clipboard operation
fastai1 copied to clipboard

Exception of pin_memory

Open giladshtern opened this issue 4 years ago • 0 comments

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 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 47, in pin_memory return data.pin_memory() RuntimeError: error in LoadLibraryA

-Gilad

giladshtern avatar Dec 08 '20 19:12 giladshtern