diff --git a/README.md b/README.md index 40c9a03..61f121e 100644 --- a/README.md +++ b/README.md @@ -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|png,id可以采用uuid,序号第一张图是1,第二张图是2,只能两张图