Мой код не работает на процессоре, пока я не установил Cuda - PullRequest
0 голосов
/ 16 октября 2019

Когда я запускаю следующий код, я получаю эту ошибку:

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 и не хочу этого делать. Как я могу запустить этот код на процессоре?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...