windows配置pytorch环境

1、安装显卡驱动

2、安装cuda

3、安装与cuda版本对应的cudnn

cudnn中三个文件夹复制到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

4、安装anaconda

注意安装时将下图中的1勾选上:
windows配置pytorch环境_第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 ./

你可能感兴趣的:(pytorch,windows,python)