Hi!
I am trying to run some code within the pytorch/pytorch:nightly-devel-cuda9.2-cudnn7 image container.
If I try to run torch.cuda.is_available() it returns true
But when I run result = model(return_loss=False, rescale=not show, **data) it throws : RuntimeError: CUDA error: no kernel image is available for execution on the device (launch_kernel at /opt/conda/conda-bld/pytorch_1579022071458/work/aten/src/ATen/native/cuda/Loops.cuh:103)
output of nvidia-smi:

image1446×680 32.8 KB

output of nvcc --version:

image842×146 11 KB

output of conda list | grep torch
pytorch                   1.4.0           py3.7_cuda9.2.148_cudnn7.6.3_0    pytorch
torchvision               0.5.0                 py37_cu92    pytorch
My Ubuntu version is 16.04
Please help!