1、安装显卡驱动
2、安装cuda
3、安装与cuda版本对应的cudnn
cudnn中三个文件夹复制到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
4、安装anaconda
注意安装时将下图中的1勾选上:
5、创建虚拟环境
6、安装apex(与linux不同)
安装apex时注意不要将apex下载到项目文件夹下。
git clone https://github.com/ptrblck/apex.git
cd apex
conda activate env_name
git checkout apex_no_distributed
pip install -v --no-cache-dir ./