fix nvidia 3000 series usage

README.md

@@ -16,16 +16,10 @@ Our paper can be downloaded from [[Arxiv]](https://arxiv.org/pdf/2106.06340v1.pd
 ## Attention
 ***This project is for technical and academic use only. Please do not apply it to illegal and unethical scenarios.***
 
-***Please do not ignore the content at the end of this README!***
-
 If you find this project useful, please star it. It is the greatest appreciation of our work.
 
 ## Top News <img width=8% src="./docs/img/new.gif"/>
 
-**`2021-11-23`**: The google drive link of [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ) is released. 
-
-**`2021-11-17`**: We released a high resolution face dataset [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ) and the method to generate this dataset. This dataset is for research purpose. 
-
 **`2021-08-30`**: Docker has been supported, please refer [here](https://replicate.ai/neuralchen/simswap-image) for details.
 
 **`2021-08-17`**: We have updated the [Preparation](./docs/guidance/preparation.md), The main change is that the gpu version of onnx is now installed by default, Now the time to process a video is greatly reduced.
@@ -42,11 +36,6 @@ If you find this project useful, please star it. It is the greatest appreciation
 
 **`2021-06-20`**: We release the scripts for arbitrary video and image processing, and a colab demo.
 
-## The first open source high resolution dataset for face swapping!!!
-## High Resolution Dataset [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ)
-
-[![logo](./docs/img/vggface2_hq_compare.png)](https://github.com/NNNNAI/VGGFace2-HQ)
-
 ## Dependencies
 - python3.6+
 - pytorch1.5+
@@ -133,7 +122,12 @@ For academic and non-commercial use only.The whole project is under the CC-BY-NC
   booktitle = {{MM} '20: The 28th {ACM} International Conference on Multimedia},
   pages     = {2003--2011},
   publisher = {{ACM}},
-  year      = {2020}
+  year      = {2020},
+  url       = {https://doi.org/10.1145/3394171.3413630},
+  doi       = {10.1145/3394171.3413630},
+  timestamp = {Thu, 15 Oct 2020 16:32:08 +0200},
+  biburl    = {https://dblp.org/rec/conf/mm/ChenCNG20.bib},
+  bibsource = {dblp computer science bibliography, https://dblp.org}
 }
 ```
 
@@ -145,19 +139,7 @@ For academic and non-commercial use only.The whole project is under the CC-BY-NC
 
 [![title](/docs/img/title.png)](https://github.com/neuralchen/ASMAGAN)
 
-**Please visit our AAAI2021 sketch based rendering project**
-
-[![logo](./docs/img/girl2.gif)](https://github.com/TZYSJTU/Sketch-Generation-with-Drawing-Process-Guided-by-Vector-Flow-and-Grayscale)
-[![title](/docs/img/girl2-RGB.png)](https://github.com/TZYSJTU/Sketch-Generation-with-Drawing-Process-Guided-by-Vector-Flow-and-Grayscale)
-
-**Please visit our high resolution face dataset VGGFace2-HQ**
-
-[![logo](./docs/img/vggface2_hq_compare.png)](https://github.com/NNNNAI/VGGFace2-HQ)
-
 Learn about our other projects 
-
-[[VGGFace2-HQ]](https://github.com/NNNNAI/VGGFace2-HQ);
-
 [[RainNet]](https://neuralchen.github.io/RainNet);
 
 [[Sketch Generation]](https://github.com/TZYSJTU/Sketch-Generation-with-Drawing-Process-Guided-by-Vector-Flow-and-Grayscale);

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.)