我是pytorch的新手。我从这个存储库https://github.com/ruotianluo/ImageCaptioning.pytorch中获取了代码,并希望为图像添加标题。安装" 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:转换issubdtype的第二个参数 不推荐
float
到np.floating
。在未来,它将是 视为np.float64 == np.dtype (float) .type
。来自._conv 将register_converters导入为_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行 = 40 error = 48:没有内核映像可用于设备Traceback(最近的最后一次呼叫最后一次):File" eval.py",line 122,在 vars(opt))文件" / home / azat /编程/ Python / techno_atom_neuro /其他实现/ ImageCaptioning.pytorch-master / eval_utils.py",第82行,in eval_split data = loader.get_batch(split)File" / home / azat / Programing / Python / techno_atom_neuro / Others Implementation / ImageCaptioning.pytorch-master / dataloaderraw.py",第112行,in get_batch img =变量(preprocess(img),volatile = True)文件" /home/azat/anaconda2/lib/python2.7/site-packages/torchvision/transforms/transforms.py", 第42行,在通话中 img = t(img)File" /home/azat/anaconda2/lib/python2.7/site-packages/torchvision/transforms/transforms.py", 第118行,在通话中 return F.normalize(tensor,self.mean,self.std)File" /home/azat/anaconda2/lib/python2.7/site-packages/torchvision/transforms/functional.py", 第161行,正常化 t.sub_(m).div_(s)RuntimeError:cuda运行时错误(48):没有内核映像可用于设备上的执行 /pytorch/torch/lib/THC/generic/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 |
+-----------------------------------------------------------------------------+
答案 0 :(得分:5)
答案隐藏在你收到的巨大错误信息中:
UserWarning:发现具有cuda功能2.1的GPU0 GeForce 820M。 PyTorch不再支持这个GPU,因为它太旧了。