Problem with Bumblebee and Pytorch

Hello there,

I have been using Bumblebee for my dGPU for a while and it has been working properly all along. I am not sure if this happened after an update or even before, but when I try to use Pytorch with CUDA it seems to be not able to find the GPU drivers.

When I use option -b none glxgears, I can see that the GPU is active and running. I can also run CUDA programs with no problem. But I can’t use the same command and run python to use Pytorch with CUDA. When I run “inxi -G” I get the following output:

       Graphics:  Device-1: Intel HD Graphics 630 driver: i915 v: kernel 
       Device-2: NVIDIA GP107M [GeForce GTX 1050 Mobile] driver: N/A 
       Device-3: Microdia Integrated_Webcam_HD type: USB driver: uvcvideo 
       Display: x11 server: X.Org 1.20.11 driver: loaded: intel unloaded: nvidia resolution: 1: 1920x1080 
       2: 1920x1080~60Hz 
       OpenGL: renderer: Mesa Intel HD Graphics 630 (KBL GT2) v: 4.6 Mesa 21.0.3 

I wanted to ask if anyone has been having the same problem.