PyTorch 0.4源码安装(Windows)
点击关注 异步图书,置顶公众号
每天与你分享 IT好书 技术干货 职场知识
在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。
需要勾选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%\include\crt\host_config.h -
#if _MSC_VER < 1600 || _MSC_VER > 1911
//把1911 改为1920即可,保存
2.1.2 安装依赖
conda设置清华大学镜像,比较快。安装pytorch 0.3也可以通过清华大学镜像
2.1.3、下载PyTorch源代码
2.2、安装
上面目录需要替换为你自己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。
参考:https://github.com/pytorch/pytorch
本文摘自异步社区:作者:RuiCylooq,《PyTorch 0.4源码安装(Windows) 》
延伸推荐
小学生开始学Python,最接近AI的编程语言:安利一波Python书单
长按二维码,可以关注我们哟
每天与你分享IT好文。
在“异步图书”后台回复“关注”,即可免费获得2000门在线视频课程;推荐朋友关注根据提示获取赠书链接,免费得异步图书一本。赶紧来参加哦!
低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。
持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。
转载内容版权归作者及来源网站所有,本站原创内容转载请注明来源。
- 上一篇
Linux中使用yum安装jdk
1.查看yum中管理的java的版本 输入以下命令,以查看可用的JDK软件包列表: yum search java | grep -i --color JDK 2.安装jdk 以java-1.8.0-openjdk-devel.x86_64为例: yum install java-1.8.0-openjdk-devel.x86_64 3.配置全局使用(可选) 让系统上的所有用户使用java(openjdk) ,则要进行下面的操作: vi /etc/profile 将下面的三行粘贴到 /etc/profile 中: export JAVA_HOME=/usr/lib/jvm/java-1.8.0-openjdk-1.8.0.161-2.b15.el7_2.x86_64 export CLASSPATH=.:$JAVA_HOME/jre/lib/rt.jar:$JAVA_HOME/lib/dt.jar:$JAVA_HOME/lib/tools.jar export PATH=$PATH:$JAVA_HOME/bin 保存关闭后,执行,让设置立即生效: source /etc/profile...
- 下一篇
Java实现一个简单的加密解密方法
Crypto是Java语言写的一个简单的加密解密方法。 使用方法: 加密方法 String cipherte=Enande.encrypt(content, pass); 解密方法 Enande.decrypt(ciphertext, pass); content,pass均为String类型 content为明文,pass为加密口令,ciphertext为密文 (对代码公开,对密码保密) /** * 时间:2007-11-14 * yster@foxmail.com */ public class Enande { /** * @param sourceString * @param password * @return 密文 */ public static String encrypt(String sourceString, String password) { char[] p = password.toCharArray(); // 字符串转字符数组 int n = p.length; // 密码长度 char[] c = sourceString.toCharArray(...
相关文章
文章评论
共有0条评论来说两句吧...
文章二维码
点击排行
推荐阅读
最新文章
- MySQL8.0.19开启GTID主从同步CentOS8
- 设置Eclipse缩进为4个空格,增强代码规范
- SpringBoot2配置默认Tomcat设置,开启更多高级功能
- SpringBoot2更换Tomcat为Jetty,小型站点的福音
- Jdk安装(Linux,MacOS,Windows),包含三大操作系统的最全安装
- CentOS8,CentOS7,CentOS6编译安装Redis5.0.7
- Springboot2将连接池hikari替换为druid,体验最强大的数据库连接池
- Hadoop3单机部署,实现最简伪集群
- CentOS8安装Docker,最新的服务器搭配容器使用
- Docker快速安装Oracle11G,搭建oracle11g学习环境