error

ERROR

[DL] RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

2023.04.25

File "/-/.conda/envs/torch38/lib/python3.8/site packages/torch/utils/data/dataloader.py", line 444, in __iter__ return self._get_iterator() 이 부분 부터 에러가 시작됐다. self.dataloader = DataLoader( dataset=self.dataset, batch_size=self.batchsize, shuffle=True, num_workers=self.num_workers, # 이 부분 추가 generator=torch.Generator(device='cuda') ) generator=torch.Generator(device='cuda') 위의 코드를 추가하여 수동으로 Genera..

ERROR

[ML] ModuleNotFoundError: No module named 'sklearn'

2023.04.17

Scikit-learn 라이브러리가 설치되지 않아서 생기는 문제로 설치해주면 된다. # 아나콘다 conda install scikit-learn #Python pip install -U scikit-learn # jupyter, colab등 sell에 설치 !pip install -U scikit-learn

[DL] RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

[ML] ModuleNotFoundError: No module named 'sklearn'

티스토리툴바