Когда я запускаю следующий код, я получаю эту ошибку:
AssertionError: Факел не скомпилирован с включенной CUDA
Я хочу запустить его на процессоре, но я нене знаешь как? Код суммирован, весь код находится по ссылке ниже:
https://github.com/Hananel-Hazan/bindsnet/blob/master/examples/mnist/reservoir.py
parser = argparse.ArgumentParser()
parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--plot", dest="plot", action="store_true")
parser.add_argument("--gpu", dest="gpu", action="store_true")
parser.add_argument("--device_id", type=int, default=0)
parser.set_defaults(plot=True, gpu=True, train=True)
args = parser.parse_args()
seed = args.seed
gpu = args.gpu
device_id = args.device_id
np.random.seed(seed)
torch.cuda.manual_seed_all(seed)
torch.manual_seed(seed)
# Sets up Gpu use
if gpu and torch.cuda.is_available():
torch.cuda.set_device(device_id)
# torch.set_default_tensor_type('torch.cuda.FloatTensor')
else:
torch.manual_seed(seed)
# Directs network to GPU
if gpu:
network.to("cuda")
# Create a dataloader to iterate and batch data
dataloader = torch.utils.data.DataLoader(
dataset, batch_size=1, shuffle=True, num_workers=0, pin_memory=gpu
)
pbar = tqdm(enumerate(dataloader))
for (i, dataPoint) in pbar:
if i > n_iters:
break
datum = dataPoint["encoded_image"].view(time, 1, 1, 28, 28).to(device_id)
model = NN(n_neurons, 10).to(device_id)
label = torch.zeros(1, 1, 10).float().to(device_id)
pbar = tqdm(enumerate(dataloader))
datum = dataPoint["encoded_image"].view(time, 1, 1, 28, 28).to(device_id)
Я не установил Cuda и не хочу этого делать. Как я могу запустить этот код на процессоре?