cupy cuda

JWDebler · Apr 28, 2021 · 7c0f46a · 7c0f46a
1 parent 4c35915
commit 7c0f46a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -32,7 +32,7 @@ RUN apt-get update && \
     find /opt/conda/ -follow -type f -name '*.js.map' -delete && \
     pip install ont-bonito && \
     bonito download --models --latest -f && \
-    pip install cupy --no-cache-dir -vvvv && \
+    pip install cupy-cuda92 && \
     apt-get autoremove --purge --yes && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*