Hi @safihaider, could you please share an example of your Dockerfile?
Here’s an example of a Dockerfile that is compatible with our GPU hardware. In general, using a docker image like FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
is greatly advised.