pytorch
只会写bug的菜鸡
这个作者很懒,什么都没留下…
展开
-
神经网络基础篇——刘二大人
#测试卷积尺寸 import torch in_channels,out_channels = 5,10 width,height = 100,100 kernel_size = 3 batch_size = 1 input = torch.randn(batch_size, in_channels, width, height) conv_layer = torch.nn...原创 2021-04-23 17:21:40 · 224 阅读 · 0 评论 -
Pytorch下载安装配置
安装看这个博客即可,写的很清楚 https://www.cnblogs.com/zhouzhiyao/p/11784055.html 测试是否安装成功 https://blog.csdn.net/weixin_38314865/article/details/102410371 import torch torch.cuda.is_available() True torch.cuda.current_device() 0 torch.cuda.device(0) <torch.cuda.转载 2020-09-29 10:01:24 · 182 阅读 · 0 评论