Windows源码安装PyTorch 0.4
在Windows安装源码安装PyTorch 0.4。
Windows 10 Home
Visual Studio 2017 Community
CUDA 9.0 + cuDNN 7.0
Anaconda 5,Python 3.6.4
1、安装软件
1.1、安装Visual Studio
需要安装Visual Studio 2017,免费的Community版是可以的,不要预览版,正式版即可。打开Visual Studio Installer,需要安装:
- 使用C++桌面开发
- VC++ 2017版本 v14.11工具箱
- Anaconda也可以勾选,这样你就可以跳过安装Anaconda这一步
安装即可。
1.2、安装Anaconda
下载Anaconda,Python3版本。
安装,勾选添加到PATH,没有勾选的话手动添加吧!
1.3、安装CUDA和cuDNN
没有英伟达显卡或者不需要英伟达显卡支持跳过,安装时需要设置环境变量NO_CUDA=1
。
我安装的是CUDA 9.0、cuDNN 7.0
需要勾选Nsight Visual Studio Edition。
2、安装PyTorch
2.1、前期准备
2.1.1 修改CUDA头文件
最新版本的Visual Studio 2017,安装时会报错,修改CUDA头文件即可:
unsupported Microsoft Visual Studio version! Only the versions 2012, 2013, 2015 and 2017 are supported!
不需要CUDA支持跳过
打开 - %CUDA_PATH%includecrthost_config.h -
#if _MSC_VER < 1600 || _MSC_VER > 1911 //把1911 改为1920即可,保存
2.1.2 安装依赖
conda create -n pytorch python=3.6 anaconda #这步不是必须的,但推荐新建一个,这样子安装失败也不会影响自己已经配置好的环境 activate pytorch #这个命令要在CMD下才有效,不知道为什么PowerShell无效 #建议此后的命令都通过CMD conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
2.1.3、下载PyTorch源代码
#保持网络比较好即可,学校网络差的要死,clone几次还不成功,最后只能用流量开个热点 git clone --recursive https://github.com/pytorch/pytorch cd pytorch
2.2、安装
xcopy /Y aten\src\ATen\common_with_cwrap.py tools\shared\cwrap_common.py set "VS150COMNTOOLS=D:\ProgramData\MicroSoft\VisualStudio\Community\VC\Auxiliary\Build" #上面目录需要替换为你自己Visual Studio安装目录 set CMAKE_GENERATOR=Visual Studio 15 2017 Win64 set DISTUTILS_USE_SDK=1 call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11 #14.13等其他版本成不成功没试过 python setup.py install
等一段时间,大功告成。我安装过程中出错就第一次忘记改CUDA的头文件引起的。
2.3、测试
先退出安装的那个CMD,重新打开一个CMD或者PowerShell。
低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。
持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。
转载内容版权归作者及来源网站所有,本站原创内容转载请注明来源。
- 上一篇
ZooKeeper监控工具(六)
之前说了那么多配置信息,具体做的怎么样,还需要真正的获取ZooKeeper的运行时信息才行。这里介绍ZooKeeper的监控。 与文无关 有如下方式: 使用内置的四字命令来获取健康状态信息 使用JMX查看运行状态 使用Exhibitor 使用Sass 四字命令 最简单的监控方式就是使用ZooKeeper的四字命令,你可以直接通过telnet或者nc命令查看状态。 四字命令 四字命令 四字命令 常用的四字命令如下: conf: 打印ZooKeeper的配置信息 cons: 列出所有的客户端会话链接 crst: 重置所有的客户端连接 dump: 打印集群的所有会话信息,包括ID,以及临时节点等信息。用在Leader节点上才有效果。 envi: 列出所有的环境参数 ruok: "谐音为Are you ok"。检查当前服务器是否正在运行。 stat: 获取ZooKeeper服务器运行时的状态信息,包括版本,运行时角色,集群节点个数等信息。 srst: 重置服务器统计信息 srvr: 和stat输出信息一样,只不过少了客户端连接信息。 wchs: 输出当前服务器上管理的Watcher概要信息 w...
- 下一篇
TensorFlow Windows 安装
欢迎大家关注我们的网站和系列教程:http://www.tensorflownews.com/,学习更多的机器学习、深度学习的知识! 本系列教程将手把手带您从零开始学习Tensorflow,并最终通过Tensorflow实现一些经典的项目。欢迎您关注我们的教程,关注更多机器学习、深度学习相关的优质博文。 Tensorflow是由谷歌大脑团队于2015年11月开发的第二代开源的机器学习系统。Tensorflow支持python、C++、java、GO等多种编程语言,以及CNN、RNN和GAN等深度学习算法。Tensorflow除可以在Windows、Linux、MacOS等操作系统运行外,还支持Android和iOS移动平台的运行、以及适用于多个CPU/GPU组成的分布式系统中。 Tensorflow是目前最火的深度学习框架,广泛应用于自然语言处理、语音识别、图像处理等多个领域。不仅深受全球深度学习爱好者的广泛欢迎,Google、eBay、Uber、OPenAI等众多科技公司的研发团队也都在使用它。 相较于其它的深度学习框架,如:Caffe、Torch、Keras、MXnet、Thean...
相关文章
文章评论
共有0条评论来说两句吧...
文章二维码
点击排行
推荐阅读
最新文章
- SpringBoot2配置默认Tomcat设置,开启更多高级功能
- SpringBoot2整合MyBatis,连接MySql数据库做增删改查操作
- SpringBoot2编写第一个Controller,响应你的http请求并返回结果
- CentOS8,CentOS7,CentOS6编译安装Redis5.0.7
- MySQL8.0.19开启GTID主从同步CentOS8
- CentOS7,CentOS8安装Elasticsearch6.8.6
- Docker使用Oracle官方镜像安装(12C,18C,19C)
- Jdk安装(Linux,MacOS,Windows),包含三大操作系统的最全安装
- Linux系统CentOS6、CentOS7手动修改IP地址
- CentOS7安装Docker,走上虚拟化容器引擎之路