Open Kinyugo opened 2 years ago
The cuda call in the training loop
while self.step < self.train_num_steps: for i in range(self.gradient_accumulate_every): data = next(self.dl).cuda() #here ....
leads to cuda error when running on CPU AssertionError: Torch not compiled with CUDA enabled.
if name == 'main': //code here
The cuda call in the training loop
leads to cuda error when running on CPU AssertionError: Torch not compiled with CUDA enabled.