From f825ec6bf464484e3ecdd7212c7e98646e131f9f Mon Sep 17 00:00:00 2001 From: Mikael Brudfors Date: Fri, 19 Nov 2021 18:11:05 +0000 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 18a2dae..eefd6aa 100644 --- a/README.md +++ b/README.md @@ -41,7 +41,7 @@ However, for running on the GPU, this only works if you ensure that the PyTorch pip install torch==1.9.0+cu111 NI_COMPILED_BACKEND="C" pip install --no-build-isolation . ``` -where the PyTorch version matches the output of `nvcc --version`. +where the PyTorch CUDA version matches the output of `nvcc --version`. ### 1.2. Example usage