
pytorch
storm岚
努力留住头发的程序员
展开
-
用pytorch搭建生成对抗网络(GAN)
生成对抗网络是2014提出的网络模型,具有很好的图像生成效果,借助最近在学习的pytorch搭建一个GAN网络,该网络采用全连接实现。 import torch from torch import nn import torchvision.transforms as tfs from torch.utils.data import DataLoader from torchvision.datasets import MNIST import numpy as np import matplotlib.p原创 2021-01-04 20:33:15 · 1958 阅读 · 2 评论 -
解决Ubuntu安装pytorch过慢的问题
最近在学习pytorch需要在服务器搭建一个安装pytorch框架,所以结合我这几天的学习经历写了一个在Ubuntu16.04版环境下安装pytorch。 1.安装anaconda anaconda是一个很好用的python包管理器,对于经常使用python是非常友好的,可以在anaconda下新建不同的环境,以满足不同开发环境的需求。anaconda的安装这个网上有很多教程,不做具体的的描述。 2.建一个anaconda环境 这是新建一个名字为pytorchgpu的环境,python版本为3.6,可根据原创 2020-12-05 19:50:39 · 2514 阅读 · 1 评论