rnakato / ShortCake

A docker image for single-cell analysis
https://hub.docker.com/r/rnakato/shortcake
GNU General Public License v3.0
66 stars 13 forks source link

Dockers jaxlib not gpu enabled #14

Open joseagraz opened 3 hours ago

joseagraz commented 3 hours ago

Dear professor Nakato, The jaxlib isn't GPU enabled. I am not 100% sure, since Dockerfile.full crashes when compiling, perhaps line #47 should be changed:

Reads: && micromamba install -n scvi-scgen-scmomat-unitvelo jax==0.3.14 jaxlib==0.1.75 -c conda-forge \

Shoudl read: && micromamba install -n scvi-scgen-scmomat-unitvelo jax[cuda11]==0.3.14 jaxlib==0.1.75 -c conda-forge \

joseagraz commented 3 hours ago

Here's the message I get: ERROR - Error doublet processing sample 3e436c7a-542e-4980-b52e-92d5d4bca496: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 803: system has unsupported display driver / cuda driver combination