PyTorch 1.0 中文文档：Windows FAQ

桃扇骨 2022-03-11 11:09 257阅读 0赞

> 译者：[冯宝宝][Link 1]

## 从源码中构建 ##

### 包含可选组件 ###

Windows PyTorch有两个受支持的组件：MKL和MAGMA。 以下是使用它们构建的步骤。

REM Make sure you have 7z and curl installed.
    
    REM Download MKL files
    curl https://s3.amazonaws.com/ossci-windows/mkl_2018.2.185.7z -k -O
    7z x -aoa mkl_2018.2.185.7z -omkl
    
    REM Download MAGMA files
    REM cuda90/cuda92/cuda100 is also available in the following line.
    set CUDA_PREFIX=cuda80
    curl -k https://s3.amazonaws.com/ossci-windows/magma_2.4.0_%CUDA_PREFIX%_release.7z -o magma.7z
    7z x -aoa magma.7z -omagma
    
    REM Setting essential environment variables
    set "CMAKE_INCLUDE_PATH=%cd%\\mkl\\include"
    set "LIB=%cd%\\mkl\\lib;%LIB%"
    set "MAGMA_HOME=%cd%\\magma"

### 为Windows构建加速CUDA ###

Visual Studio当前不支持并行自定义任务。 作为替代方案，我们可以使用Ninja来并行化CUDA构建任务。 只需键入几行代码即可使用它。

REM Let's install ninja first.
    pip install ninja
    
    REM Set it as the cmake generator
    set CMAKE_GENERATOR=Ninja

> [**阅读全文／改进本文**][Link 2]

[Link 1]: https://github.com/PEGASUS1993
[Link 2]: https://github.com/apachecn/pytorch-doc-zh/blob/master/docs/1.0/notes_windows.md

发表评论取消回复

表情：

评论列表（有 0 条评论，257人围观）

还没有评论，来说两句吧...

相关阅读

相关 PyTorch 中文文档 - 教程

PyTorch 中文文档 - 教程 PyTorch 中文文档 [https://www.pytorchtutorial.com/docs/][https_www.p

「爱情、让人受尽委屈。」/ 2023年02月16日 04:14/ 0 赞/ 8 阅读

相关 PyTorch 1.0 中文文档：torchvision.models

> 译者：[BXuan694][] models子包定义了以下模型架构： [AlexNet][] [VGG][] [ResNet][] [Squee

素颜马尾好姑娘i/ 2022年03月11日 11:26/ 0 赞/ 255 阅读

相关 PyTorch 1.0 中文文档：torchvision 参考

> 译者：[BXuan694][] `torchvision` 包收录了若干重要的公开数据集、网络模型和计算机视觉中的常用图像变换包参考 torchvision.

柔情只为你懂/ 2022年03月11日 11:26/ 0 赞/ 222 阅读

相关 PyTorch 1.0 中文文档：torch.hub

> 译者：[kunwuz][] torch.hub.load(github, model, force_reload=False, args, kwargs) 从g

朱雀/ 2022年03月11日 11:24/ 0 赞/ 676 阅读

相关 PyTorch 1.0 中文文档：Torch 脚本

> 译者：[keyianpai][] 创建 Torch 脚本代码将追踪和脚本化结合起来 Torch 脚本语言参考类型

待我称王封你为后i/ 2022年03月11日 11:14/ 0 赞/ 221 阅读

相关 PyTorch 1.0 中文文档：torch.distributions

> 译者：[hijkzzz][] `distributions` 包含可参数化的概率分布和采样函数. 这允许构造用于优化的随机计算图和随机梯度估计器. 这个包一般遵循 [Te

Myth丶恋晨/ 2022年03月11日 11:12/ 0 赞/ 261 阅读

相关 PyTorch 1.0 中文文档：torch.distributed

> 译者：[univeryinli][] 后端 `torch.distributed` 支持三个后端，每个后端具有不同的功能。下表显示哪些功能可用于CPU/CUDA张量

左手的ㄟ右手/ 2022年03月11日 11:12/ 0 赞/ 261 阅读

相关 PyTorch 1.0 中文文档：torch.autograd

> 译者：[gfjiangly][] `torch.autograd` 提供类和函数，实现任意标量值函数的自动微分。它要求对已有代码的最小改变—你仅需要用`requires

冷不防/ 2022年03月11日 11:09/ 0 赞/ 267 阅读

相关 PyTorch 1.0 中文文档：Windows FAQ

> 译者：[冯宝宝][Link 1] 从源码中构建包含可选组件 Windows PyTorch有两个受支持的组件：MKL和MAGMA。以下是使用它们构建的步骤

桃扇骨/ 2022年03月11日 11:09/ 0 赞/ 258 阅读

相关 PyTorc 1.0 中文文档：扩展PyTorch

> 译者：[PEGASUS1993][] 本章中，将要介绍使用我们的C库如何扩展`torch.nn`，`torch.autograd`和编写自定义的`C`扩展工具。扩展

分手后的思念是犯贱/ 2022年03月11日 11:07/ 0 赞/ 204 阅读