From 63fb5fc46fa8e484cb6b02ae10283e245db4cd91 Mon Sep 17 00:00:00 2001 From: Tijs Zwinkels Date: Tue, 16 May 2023 12:07:09 +0200 Subject: [PATCH] Suggest using pytorch-cuda 11.8 instead of 11.7 This prevents CuFFT errors on newer cards such as the RTX 4090 and RTX 6000 Ada. fixes #254 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index a660d2d..28345f1 100644 --- a/README.md +++ b/README.md @@ -74,7 +74,7 @@ GPU execution requires the NVIDIA libraries cuBLAS 11.x and cuDNN 8.x to be inst ### 2. Install PyTorch2.0, e.g. for Linux and Windows CUDA11.7: -`conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia` +`conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia` See other methods [here.](https://pytorch.org/get-started/previous-versions/#v200)