Pytorch CUDA error: no kernel image is available for execution on the device on RTX 3090 with cuda 11.1

Question

If I run the following:

import torch
import sys
print('A', sys.version)
print('B', torch.__version__)
print('C', torch.cuda.is_available())
print('D', torch.backends.cudnn.enabled)
device = torch.device('cuda')
print('E', torch.cuda.get_device_properties(device))
print('F', torch.tensor([1.0, 2.0]).cuda())

I get this:

A 3.7.5 (default, Nov  7 2019, 10:50:52) 
[GCC 8.3.0]
B 1.8.0.dev20210115+cu110
C True
D True
E _CudaDeviceProperties(name='GeForce RTX 3090', major=8, minor=6, total_memory=24267MB, multi_processor_count=82)
F 
<stacktrace>
CUDA error: no kernel image is available for execution on the device

More info about my system:

Nvidia version: NVIDIA-SMI 455.38 Driver Version: 455.38 CUDA Version: 11.1
python 3.7, Ubuntu 18.04

Benjamin Crouzier · Accepted Answer · 2021-01-15 16:21:29Z

12

Found a fix for my problem here: https://github.com/pytorch/pytorch/issues/31285#issuecomment-739139454

pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu110/torch_nightly.html -U

Then my code snippet gives:

A 3.7.5 (default, Nov  7 2019, 10:50:52) 
[GCC 8.3.0]
B 1.8.0.dev20210115+cu110
C True
D True
E _CudaDeviceProperties(name='GeForce RTX 3090', major=8, minor=6, total_memory=24267MB, multi_processor_count=82)
F tensor([1., 2.], device='cuda:0')

answered Jan 15, 2021 at 16:21

Benjamin Crouzier

42.2k48 gold badges179 silver badges240 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

mjeday Over a year ago

still doesn't work

Om Rastogi · Accepted Answer · 2022-05-05 04:40:48Z

1

https://pytorch.org/get-started/locally

Use this page to select the configuration of the system. And run the command that is displayed. It works in all kinds of installation problems.

answered May 5, 2022 at 4:40

Om Rastogi

1,0673 gold badges11 silver badges16 bronze badges

Collectives™ on Stack Overflow

Pytorch CUDA error: no kernel image is available for execution on the device on RTX 3090 with cuda 11.1

2 Answers 2

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related