waspinator
waspinator
PostsComments
RRunPod
Created by waspinator on 6/4/2024 in #⛅｜pods-clusters
how do you create a compatible docker file?
when I use FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04 I get this error when trying to use the gpu locally. 
UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 804: forward compatibility was attempted on non supported HW (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:109.)
UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 804: forward compatibility was attempted on non supported HW (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:109.)
I don't get that error when using the official pytorch image. Have you seen that before?
20 replies