-
Notifications
You must be signed in to change notification settings - Fork 345
Description
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