windows配置pytorch环境

1、安装显卡驱动

2、安装cuda

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

cudnn中三个文件夹复制到路径C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.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 ./

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>