Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RuntimeError: CUDA error: no kernel image is available for execution on the device #174

Open
kriti1307 opened this issue Feb 8, 2024 · 0 comments

Comments

@kriti1307
Copy link

Hii, I am facing issue realted to CUDA. after trying to test.py file. I am using the mentioned compatible driver in linux version 22.04.
Screenshot from 2024-02-08 15-08-28
Can i have to dowgrade Cuda? or I have to downgrade the linux ? and i am also confused which cuda version support for which linux version?
The error is :
....anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/cuda/init.py:104: UserWarning:
NVIDIA GeForce RTX 3050 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 compute_37.
If you want to use the NVIDIA GeForce RTX 3050 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
model [FaceReconModel] was created
loading the model from ./checkpoints/facemodel/epoch_20.pth
0 ./datasets/examples/000002.jpg
Traceback (most recent call last):
File "test.py", line 73, in
main(0, opt,opt.img_folder)
File "test.py", line 63, in main
model.test() # run inference
File "/home/kriti/Desktop/Deep3DFaceRecon_pytorch/models/base_model.py", line 162, in test
self.forward()
File "/home/kriti/Desktop/Deep3DFaceRecon_pytorch/models/facerecon_model.py", line 135, in forward
output_coeff = self.net_recon(self.input_img)
File "/home/kriti/anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/kriti/Desktop/Deep3DFaceRecon_pytorch/models/networks.py", line 98, in forward
x = self.backbone(x)
File "/home/kriti/anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/kriti/Desktop/Deep3DFaceRecon_pytorch/models/networks.py", line 375, in forward
return self._forward_impl(x)
File "/home/kriti/Desktop/Deep3DFaceRecon_pytorch/models/networks.py", line 360, in _forward_impl
x = self.relu(x)
File "/home/kriti/anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in call_impl
result = self.forward(*input, **kwargs)
File "/home/kriti/anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 102, in forward
return F.relu(input, inplace=self.inplace)
File "/home/kriti/anaconda3/envs/deep3d_pytorch_new/lib/python3.6/site-packages/torch/nn/functional.py", line 1134, in relu
result = torch.relu
(input)
RuntimeError: CUDA error: no kernel image is available for execution on the device

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant