Support FP32 (#141)

This commit is contained in:
Woosuk Kwon authored and GitHub committed 2023-06-07 00:40:21 -07:00
1 parent 376725ce74
commit e38074b1e6
8 files changed
+65 -54

No files matched your search

@@ -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
----------------