深度学习
Skyline_98
积跬步,至千里。
展开
-
Loaded runtime CuDNN library: 7605 (compatibility version 7600) but source was compiled with 7003
问题原因:当前环境的cudnn版本(v7.6)与代码编写的cudnn版本(v7.0)不匹配。解决办法:安装cudnn v7.0。在使用anaconda安装cudnn时,发现国内镜像网站中都没有cudnn的7.0版本,可能是因为7.0版本实在太老了。其实可以直接从官网下载cudnn v7.0的库,然后手动替换到当前的anaconda环境中,具体步骤如下:1、在Nvidia的官网下载cudnn v7.0;2、解压cudnn v7.0压缩包,将文件夹中的bin、lib以及include文件夹复制到anac原创 2021-04-15 11:17:39 · 1200 阅读 · 3 评论 -
找不到‘_pywrap_tensorflow_internal‘模块
原因:TensorFlow版本太高了,与代码不匹配。解决方法:安装版本低于tensorflow1.7.1的版本。原创 2021-04-12 09:49:16 · 782 阅读 · 0 评论 -
解决问题:ModuleNotFoundError: No module named ‘__main__.XX‘; ‘__main__‘ is not a package
同一个项目里面,导入自己写的模块时,出现“ModuleNotFoundError: No module named ‘main.XX’; ‘main’ is not a package”问题。 这是因为import时使用了相对路径导致的,把相对路径改为绝对路径可以解决这个问题。这个问题可能与pycharm版本有关。...原创 2021-04-09 15:09:32 · 2853 阅读 · 0 评论 -
PyTorch入门——使用GPU并行处理数据
使用PyTorch我们可以方便的将数据及神经网络传递到GPU,使用GPU来加速网络训练。1、关键函数:# 将运行设备设置为GPUdevice = torch.device("cuda:0")# 将神经网络传递给GPUmodel.to(device)# 将Tensor数据传递给GPUmytensor = my_tensor.to(device)# 设置并行处理数据model = nn.DataParallel(model)2、使用G原创 2021-01-07 21:43:03 · 1396 阅读 · 2 评论 -
PyTorch入门——训练图像分类器
本文将使用PyTorch搭建神经网络,训练一个简单的图像分类器,主要分为以下步骤:1. 使用torchvision对CIFAR10训练和测试数据集进行加载和归一化import torchimport torchvisionimport torchvision.transforms as transforms####################################################################原创 2021-01-07 21:27:12 · 433 阅读 · 0 评论 -
PyTorch入门——搭建神经网络
使用PyTorch搭建并训练神经网络主要分为以下五个步骤:1. 定义神经网络; 使用PyTorch只需要定义神经网络的正向传输过程,Autograd机制会自动生成反向传播过程。torch.nn包提供了创建神经网络的卷积层、全连接层等的函数,使用这些函数可以方便的搭建想要的网络。代码示例:import torchimport torch.nn as nnimport torch.nn.fu原创 2021-01-07 21:15:09 · 337 阅读 · 0 评论 -
PyTorch入门——Autograd
一、什么是Autograd Autograd是PyTorch中所有神经网络的核心,它提供自动计算张量梯度的方法。使用Autograd我们在搭建神经网络是只需要定义正向传播过程,PyTorch会自动生成反向传播过程的计算公式。二、如何使用Autograd Tensor是Autograd的核心类,我们在使用时要把需要自动计算梯度的Tensor的requires_grad设为True,把不需要计算梯度原创 2021-01-07 20:49:04 · 1083 阅读 · 0 评论 -
PyTorch入门——什么是Pytorch?
一、什么是Pytorch Pytorch是以一个基于Python的科学计算包,其目标有两个:替代Numpy使用GPU来处理数据;提供最大灵活性和速度的深度学习研究平台。二、如何使用Pytorch处理数据 Pytorch以Tensor(张量)来组织数据,张量类似于数组,不同的是张量可以使用GPU来加速计算。 下面从五个方面介绍Tens原创 2021-01-07 20:06:30 · 1048 阅读 · 0 评论