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
二维码