Support FP32 (#141)

2023-06-07 00:40:21 -07:00
parent 376725ce74
commit e38074b1e6
8 changed files with 65 additions and 54 deletions
--- a/docs/source/getting_started/installation.rst
+++ b/docs/source/getting_started/installation.rst
@@ -18,8 +18,11 @@ CacheFlow can run on systems that meet the following requirements:

    .. code-block:: console

+        $ # Pull the Docker image with CUDA 11.8.
        $ docker run --gpus all -it --rm --shm-size=8g nvcr.io/nvidia/pytorch:22.12-py3

+    Inside the Docker container, please execute :code:`pip uninstall torch` before installing CacheFlow.
+
 Install with pip
 ----------------