Pytorch使用GPU

最新推荐文章于 2024-08-03 16:13:05 发布

小川爱学习

最新推荐文章于 2024-08-03 16:13:05 发布

阅读量2.1w

点赞数 4

分类专栏：计算机视觉文章标签： pytorch

本文链接：https://blog.csdn.net/wuichuan/article/details/66969315

版权

本文详细介绍了如何在Pytorch中利用GPU进行深度学习，包括数据准备、模型定义、训练过程，以及如何将模型和数据迁移到GPU上进行计算，强调了Pytorch在GPU加速方面的简便性和高效性。

摘要由CSDN通过智能技术生成

pytorch如何使用GPU

在本文中，我将介绍简单如何使用GPU
pytorch是一个非常优秀的深度学习的框架，具有速度快，代码简洁，可读性强的优点。
我们使用pytorch做一个简单的回归。
首先准备数据

import numpy as np
import matplotlib.pyplot as plt
import torch
from torch.autograd import  Variable
import torch.nn as nn
x = np.random.randn(1000, 1)*4
w = np.array([0.5,])
bias = -1.68

y_true = np.dot(x, w) + bias  #真实数据
y = y_true + np.random.randn(x.shape[0])#加噪声的数据
#我们需要使用x和y，以及y_true回归出w和bias