Windows源码安装PyTorch 0.4

简介: 在Windows源码安装Pytorch0.4,CUDA支持

在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这一步

勾选使用C++桌面开发
VC++ 2017版本 v14.11工具箱
安装即可。

1.2、安装Anaconda

下载Anaconda,Python3版本。
安装,勾选添加到PATH,没有勾选的话手动添加吧!
勾选添加到PATH变量

1.3、安装CUDA和cuDNN

没有英伟达显卡或者不需要英伟达显卡支持跳过,安装时需要设置环境变量NO_CUDA=1
我安装的是CUDA 9.0cuDNN 7.0
需要勾选Nsight Visual Studio Edition
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。
测试

参考:https://github.com/pytorch/pytorch

相关文章
|
1月前
|
并行计算 开发工具 异构计算
在Windows平台使用源码编译和安装PyTorch3D指定版本
【10月更文挑战第6天】在 Windows 平台上,编译和安装指定版本的 PyTorch3D 需要先安装 Python、Visual Studio Build Tools 和 CUDA(如有需要),然后通过 Git 获取源码。建议创建虚拟环境以隔离依赖,并使用 `pip` 安装所需库。最后,在源码目录下运行 `python setup.py install` 进行编译和安装。完成后即可在 Python 中导入 PyTorch3D 使用。
197 0
|
5月前
|
Apache Windows
windows源码安装apache2 win安装apache 阿帕奇服务器
windows源码安装apache2 win安装apache 阿帕奇服务器
170 0
|
6月前
|
PyTorch 算法框架/工具 C++
windows上编译安装pytorch的c++扩展
windows上编译安装pytorch的c++扩展
115 0
|
并行计算 PyTorch Linux
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
6680 0
|
并行计算 PyTorch Linux
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南
3756 0
|
机器学习/深度学习 人工智能 Linux
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程
|
机器学习/深度学习 并行计算 PyTorch
Windows系统下有英伟达显卡安装PyTorch
Windows系统下有英伟达显卡安装PyTorch
166 0
|
机器学习/深度学习 并行计算 PyTorch
PyTorch无GPU加速版本windows笔记
PyTorch无GPU加速版本windows笔记
215 0
PyTorch无GPU加速版本windows笔记
|
机器学习/深度学习 并行计算 PyTorch
【Pytorch-入门】windows+Anaconda虚拟环境搭建(避免不必要的包冲突)
【Pytorch-入门】windows+Anaconda虚拟环境搭建(避免不必要的包冲突)
525 0
【Pytorch-入门】windows+Anaconda虚拟环境搭建(避免不必要的包冲突)
|
PyTorch 编译器 算法框架/工具
基于Windows中学习Deep Learning之搭建Anaconda+Cudnn+Cuda+Pytorch+Pycharm工具和配置环境完整最简版(二)
基于Windows中学习Deep Learning之搭建Anaconda+Cudnn+Cuda+Pytorch+Pycharm工具和配置环境完整最简版
146 0
基于Windows中学习Deep Learning之搭建Anaconda+Cudnn+Cuda+Pytorch+Pycharm工具和配置环境完整最简版(二)

热门文章

最新文章