240 发简信
IP属地:陕西
  • 120
    基于PyTorch的CIFAR10小记

    CIFAR-10数据集介绍 CIFAR-10数据集由10个类的60000个32x32彩色图像组成,每个类有6000个图像。有50000个训练图像和10000个测试图像。数据集...

  • “Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法

    最近在使用TF2.0,运行程序时出现以下错误: 一开始怀疑是CUDA和CuDNN配置错误。反复试验后发现可能是GPU内存不足造成的。需要在程序前加以下一段代码: 意思是对GP...