R
RunPod•4d ago
canxerian

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda

Hey everyone 👋 I'm trying to use Runpod serverless to run the https://github.com/nerfstudio-project/gsplat/ gaussian-splatting implementation. However, when building the project from source (pip install . of my Dockerfile below), I get the error: No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda Is the CUDA runtime available during the build stage of a docker image on runpod serverless workers? Thanks in advance! Dockerfile:
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04

WORKDIR /

# Python dependencies
COPY builder/requirements.txt /requirements.txt
RUN python -m pip install --upgrade pip && \
python -m pip install --upgrade -r /requirements.txt --no-cache-dir && \
rm /requirements.txt


# Install gsplat deps
RUN apt update && apt install libglm-dev g++-9 gcc-9 -y
RUN python -m pip install ninja numpy jaxtyping rich

# Install gsplat
RUN git clone https://github.com/nerfstudio-project/gsplat.git && \
cd gsplat && \
python -m pip install . && \
cd examples && \
python -m pip install -r requirements.txt
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04

WORKDIR /

# Python dependencies
COPY builder/requirements.txt /requirements.txt
RUN python -m pip install --upgrade pip && \
python -m pip install --upgrade -r /requirements.txt --no-cache-dir && \
rm /requirements.txt


# Install gsplat deps
RUN apt update && apt install libglm-dev g++-9 gcc-9 -y
RUN python -m pip install ninja numpy jaxtyping rich

# Install gsplat
RUN git clone https://github.com/nerfstudio-project/gsplat.git && \
cd gsplat && \
python -m pip install . && \
cd examples && \
python -m pip install -r requirements.txt
GitHub
GitHub - nerfstudio-project/gsplat: CUDA accelerated rasterization ...
CUDA accelerated rasterization of gaussian splatting - nerfstudio-project/gsplat
5 Replies
nerdylive
nerdylive•3d ago
is cuda 11.8.0 supported or which cuda should your app / program use? (for that gsplat app)
canxerian
canxerianOP•3d ago
It supports CUDA 11.8. I've ran the same installation process on a pod using the runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04 template and it works
nerdylive
nerdylive•3d ago
oh maybe a bad pod?, it works on pod, but it doesn't in a serverless whats your endpoint id? open a ticket here
Poddy
Poddy•3d ago
@canxerian
Escalated To Zendesk
The thread has been escalated to Zendesk!
canxerian
canxerianOP•3d ago
Thanks @nerdylive! Endpoint ID is pr3gt0gti2ylyo

Did you find this page helpful?