Open baiziyuandyufei opened 2 years ago
Same problem here, with Cuda 11.6 and torch 1.12 (and trankit 1.1.1)
Same error here
trankit.Pipeline(lang="german-hdt", gpu=True, cache_dir="./cache")
=> RuntimeError: CUDA error: no kernel image is available for execution on the device
torch.cuda.is_available()
returns True
when I run
p=Pipeline('auto')
docker image is nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu18.04
$ nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Sun_Jul_28_19:07:16_PDT_2019 Cuda compilation tools, release 10.1, V10.1.243
$ nvidia-smi NVIDIA-SMI 470.103.01 Driver Version: 470.103.01 CUDA Version: 11.4
$ pip list | grep torch
torch 1.11.0