From fd090b8999a3d089346dc03e5316ac088a980df8 Mon Sep 17 00:00:00 2001
From: realies <5107843+realies@users.noreply.github.com>
Date: Wed, 22 Sep 2021 02:18:55 +0300
Subject: [PATCH] fix nvidia 3000 series usage

---
 docs/guidance/preparation.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/guidance/preparation.md b/docs/guidance/preparation.md
index dc7f766a..14aa0092 100644
--- a/docs/guidance/preparation.md
+++ b/docs/guidance/preparation.md
@@ -6,7 +6,7 @@
 ```
 conda create -n simswap python=3.6
 conda activate simswap
-conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
+conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c nvidia -c conda-forge
 (option): pip install --ignore-installed imageio
 pip install insightface==0.2.1 onnxruntime moviepy
 (option): pip install onnxruntime-gpu  (If you want to reduce the inference time)(It will be diffcult to install onnxruntime-gpu , the specify version of onnxruntime-gpu may depends on your machine and cuda version.)