我是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 = 40 error = 48:没有内核映像可用于设备Traceback(最近一次最后一次调用):文件“eval.py”,第122行,在vars(opt))文件“/ home / azat / Programing / Python / techno_atom_neuro / Others Implementation / ImageCaptioning.pytorch-master / eval_utils.py“,第82行,在eval_split中data = loader.get_batch(split)文件”/ home / azat / Programing / Python / techno_atom_neuro / Others实现/ ImageCaptioning.pytorch-master / dataloaderraw.py“,第112行,在get_batch img = Variable(preprocess(img),volatile = True)文件”/home/azat/anaconda2/lib/python2.7/site-packages/torchvision /transforms/transforms.py“,第42行,在调用img = t(img)文件”/home/azat/anaconda2/lib/python2.7/s ite-packages / torchvision / transforms / transforms.py“,第118行,在调用返回F.normalize(tensor,self.mean,self.std)文件”/home/azat/anaconda2/lib/python2.7/site- packages / torchvision / transforms / functional.py“,第161行,标准化t.sub(m).div(s)RuntimeError:cuda运行时错误(48):没有内核映像可以在/ pytorch /上的设备上执行火炬/ lib目录/ THC /通用/ THCTensorMathPairwise.cu:40
所以,我想知道这个错误是什么,硬件或软件 . 我该如何解决这个问题呢 . 谢谢 .
PyTorch
OS: Ubuntu 16.04
PyTorch version: 0.3.1 pip
Python version: Python 2.7.14 :: Anaconda custom (64-bit)
CUDA/cuDNN version: 9.1
GPU models and configuration:
$ nvidia-smi
Wed Apr 11 21:34:08 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.48 Driver Version: 390.48 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce 820M Off | 00000000:08:00.0 N/A | N/A |
| N/A 54C P0 N/A / N/A | 114MiB / 1985MiB | N/A Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 Not Supported |
+-----------------------------------------------------------------------------+
1 回答
答案隐藏在你得到的巨大错误信息中: