记录一下在Windows 下编译 PyTorch 0.5的经历吧~

本文详细记录了在Windows下编译PyTorch 0.5的过程,包括所需软件版本(如VS2017、Anaconda、CUDA 9.1、cuDNN 7.1、cmake等)和注意事项,如VS2017需选15.4工具集,CUDA不支持最新编译器,以及解决nvcc错误和Python环境问题。通过conda环境管理和脚本编译,最终成功完成编译。
摘要由CSDN通过智能技术生成

首先列出一下参考的文章地址:

关于Windows PRs并入PyTorch的master分支——蒲嘉宸

还有作者的GitHub链接:peterjc123/pytorch-scripts

非常感谢作者的分享~~~

+++++++++++++++分割线+++++++++++++++++++++++++++++++

前期准备:VS2017 、Anaconda 、CUDA 9.1 、cuDNN 7.1、cmake 3.11.2 、MKL 

PS:VS2017 安装的时候一定要选 15.4 V14.11工具集,CUDA 9.1 不支持最新的编译器。

PS1:CUDA最好只保留最新的,或者留一下,如果PATH 下存在多个CUDA_PATH_V(版

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值