Update README.md

This commit is contained in:
wlkjyy 2023-10-21 17:11:07 +08:00 committed by GitHub
parent a9c6955d96
commit a65964454e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,13 +19,32 @@
#### 环境安装
安装环境我用到的是python3.10
```
conda create -n geetest python=3.10
conda create -n geetest python=3.8
```
安装必要的环境
```
pip3 install -r requirement.txt
```
### CUDA安装
换conda源
```
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes
```
安装cuda和 cudnn
```
conda install cudnn=8.1.0.77 cudatoolkit=11.2.0
```
#### 数据准备
准备数据集放入data中格式为 id_序号.jpg|pngid可以采用uuid序号第一张图是1第二张图是2只能两张图