pytorch和keras测试gpu是否可以调用

pytorch

import numpy as np
import torch
x = np.linspace(0,20,100).astype(np.float32).reshape((-1,1))
y = x*4 + np.random.rand((*x.shape)).astype(np.float32).reshape((-1,1))


import torch.nn as nn
class LinearRegressionModel(nn.Module):

    def __init__(self, input_dim, output_dim):

        super(LinearRegressionModel, self).__init__() 
        self.linear = nn.Linear(input_dim, output_dim)

    def forward(self, x):

        out = self.linear(x)
        return out

input_dim = 1
output_dim = 1

net = LinearRegressionModel(input_dim,output_dim)
net = net.cuda()

criterion = nn.MSELoss()# Mean Squared Loss
l_rate = 0.001
optimizer = torch.optim.SGD(net.parameters(), lr = l_rate) #Stochastic Gradient Descent

epochs = 10000


for epoch in range(epochs):

    epoch +=1
    #increase the number of epochs by 1 every time
    inputs = torch.from_numpy(x)
    labels = torch.from_numpy(y)
    inputs = inputs.cuda()
    labels = labels.cuda()
    outputs = net(inputs)
    
    optimizer.zero_grad()
    loss = criterion(outputs, labels)
    loss.backward()
    optimizer.step()
    
    if epoch % 200 == 0:
        print('epoch {}, loss {}, w {}, b {},'.format(epoch,loss.item(), net.linear.weight.item(), net.linear.bias.item()))

keras

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# Copyright © 2020 2020-08-10 carryhjr
#
# Distributed under terms of the MIT license.

import numpy as np
np.random.seed(1337)  
from keras.models import Sequential
from keras.layers import Dense
import matplotlib.pyplot as plt
 
X = np.linspace(-1, 1, 200)
np.random.shuffle(X)   
Y = 0.5 * X + 2 + np.random.normal(0, 0.05, (200, )) 
# plot
plt.scatter(X, Y)
plt.show()
 
X_train, Y_train = X[:160], Y[:160]    
X_test, Y_test = X[160:], Y[160:]     
 

model = Sequential()
model.add(Dense(input_dim=1, units=1))
 

model.compile(loss='mse', optimizer='sgd')
 

print('Training -----------')
for step in range(501):
    cost = model.train_on_batch(X_train, Y_train)
    if step % 50 == 0:
        print("After %d trainings, the cost: %f" % (step, cost))

print('\nTesting ------------')
cost = model.evaluate(X_test, Y_test, batch_size=40)
print('test cost:', cost)
W, b = model.layers[0].get_weights()