mamba icon indicating copy to clipboard operation
mamba copied to clipboard

Published 20 hours ago • verified publisher icon state-spaces

The GPU utilization error。

Open CYYJL opened this issue 10 months ago • 0 comments

Hi，When I run Mamba in a Docker container, GPU 0 is running YOLOv8, while GPU 1 is running Mamba. However, the GPU utilization of GPU 1 is 0, while GPU 0's utilization is normal. I don't know how to solve this problem and hope to get your advice. I am using torch 2.1.1, CUDA 11.8, Mamba_ssm 1.0.1, Causal-conv1d 1.0.0, and Ubuntu 22.04 system. 1713344485(1)

CYYJL avatar Apr 17 '24 09:04 CYYJL