![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Bokman
这个作者很懒,什么都没留下…
展开
-
python中plt.imshow()不显示图片
直接利用plt.imshow()发现居然不能显示图片,最后加了plt.show()发现还是不能显示 解决方法 首先导入pylab包 import pylab 然后在plt.imshow(img)后面添加一行代码 pylab.show() 再次运行即可查看到图片转载 2020-11-07 16:49:18 · 1145 阅读 · 2 评论 -
查看pytorch版本
新建.py文件并输入以下代码: import torch print(torch.__version__)原创 2020-11-07 16:45:46 · 262 阅读 · 0 评论 -
pytorch 1.7.0 读取mnist数据集加载并显示图像
完整源代码: import torch import torchvision import torchvision.transforms as transforms from matplotlib import pyplot import matplotlib.pyplot as plt import numpy as np batch_size = 64 train_dataset = torchvision.datasets.MNIST(root='data', train=True, tran原创 2020-11-07 09:57:25 · 1064 阅读 · 0 评论