我是pytorch的新手 . 我从这个存储库https://github.com/ruotianluo/ImageCaptioning.pytorch获取代码,并希望为图像制作 Headers . 安装"CUDA",当我运行脚本来创建注释时,会产生以下结果:
$ CUDA_LAUNCH_BLOCKING = 1 python eval.py --model model.pth - -infos_path infos.pkl --image_folder blah --num_images 1 /home/azat/anaconda2/lib/python2.7/site-packages/h5py/init . py:36:FutureWarning:不推荐使用float tonp.floating转换issubdtype的第二个参数 . 将来,它将被视为np.float64 == np.dtype(float).type . 从.conv import register_converters as register_converters /home/azat/anaconda2/lib/python2.7/site-packages/torch/cuda/init.py:97:UserWarning:发现GPU0 GeForce 820M具有cuda功能2.1 . PyTorch不再支持这款GPU,因为它太旧了 . warnings.warn(old_gpu_warn%(d,name,major,capability 1))DataLoaderRaw从文件夹加载图像:blah 0列出目录中的所有图像blah DataLoaderRaw找到8张图像THCudaCheck FAIL文件= / pytorch / torch / lib / THC / generic / THCTensorMathPairwise.cu line