Lines Matching +full:build +full:- +full:docker +full:- +full:cuda
1  with strong GPU acceleration
7 - Deep neural networks built on a tape-based autograd system
13 <!-- toc -->
15 - [More About PyTorch](#more-about-pytorch)
16 - [A GPU-Ready Tensor Library](#a-gpu-ready-tensor-library)
17 - [Dynamic Neural Networks: Tape-Based Autograd](#dynamic-neural-networks-tape-based-autograd)
18 - [Python First](#python-first)
19 - [Imperative Experiences](#imperative-experiences)
20 - [Fast and Lean](#fast-and-lean)
21 - [Extensions Without Pain](#extensions-without-pain)
22 - [Installation](#installation)
23 - [Binaries](#binaries)
24 - [NVIDIA Jetson Platforms](#nvidia-jetson-platforms)
25 - [From Source](#from-source)
26 - [Prerequisites](#prerequisites)
27 - [NVIDIA CUDA Support](#nvidia-cuda-support)
28 - [AMD ROCm Support](#amd-rocm-support)
29 - [Intel GPU Support](#intel-gpu-support)
30 - [Get the PyTorch Source](#get-the-pytorch-source)
31 - [Install Dependencies](#install-dependencies)
32 - [Install PyTorch](#install-pytorch)
33 - [Adjust Build Options (Optional)](#adjust-build-options-optional)
34 - [Docker Image](#docker-image)
35 - [Using pre-built images](#using-pre-built-images)
36 - [Building the image yourself](#building-the-image-yourself)
37 - [Building the Documentation](#building-the-documentation)
38 - [Previous Versions](#previous-versions)
39 - [Getting Started](#getting-started)
40 - [Resources](#resources)
41 - [Communication](#communication)
42 - [Releases and Contributing](#releases-and-contributing)
43 - [The Team](#the-team)
44 - [License](#license)
46 <!-- tocstop -->
55 | ---- | --- |
57 | [**torch.autograd**](https://pytorch.org/docs/stable/autograd.html) | A tape-based automatic diff…
65 - A replacement for NumPy to use the power of GPUs.
66 - A deep learning research platform that provides maximum flexibility and speed.
70 ### A GPU-Ready Tensor Library
83 ### Dynamic Neural Networks: Tape-Based Autograd
88 One has to build a neural network and reuse the same structure again and again.
91 With PyTorch, we use a technique called reverse-mode auto-differentiation, which allows you to
94 [torch-autograd](https://github.com/twitter/torch-autograd),
107 …tps://www.numpy.org/) / [SciPy](https://www.scipy.org/) / [scikit-learn](https://scikit-learn.org)…
140 [or your favorite NumPy-based libraries such as SciPy](https://pytorch.org/tutorials/advanced/numpy…
143 …rials/advanced/cpp_extension.html) and [an example here](https://github.com/pytorch/extension-cpp).
149 …pip wheels are on our website: [https://pytorch.org/get-started/locally/](https://pytorch.org/get-…
154 …rch-for-jetson-version-1-10-now-available/72048) and the L4T container is published [here](https:/…
156 They require JetPack 4.2 and above, and [@dusty-nv](https://github.com/dusty-nv) and [@ptrblck](htt…
163 - Python 3.8 or later (for Linux, Python 3.8.1+ is needed)
164 - A compiler that fully supports C++17, such as clang or gcc (gcc 9.4.0 or newer is required, on Li…
165 - Visual Studio or Visual Studio Build Tool on Windows
168 Professional, or Community Editions. You can also install the build tools from
169 https://visualstudio.microsoft.com/visual-cpp-build-tools/. The build tools *do not*
172 …naconda](https://www.anaconda.com/download) environment. You will get a high-quality BLAS library …
180 $ conda create -y -n <CONDA_NAME>
188 $ conda create -y -n <CONDA_NAME>
190 $ call "C:\Program Files\Microsoft Visual Studio\<VERSION>\Community\VC\Auxiliary\Build\vcvarsall.b…
193 ##### NVIDIA CUDA Support
194 …f you want to compile with CUDA support, [select a supported version of CUDA from our support matr…
195 - [NVIDIA CUDA](https://developer.nvidia.com/cuda-downloads)
196 - [NVIDIA cuDNN](https://developer.nvidia.com/cudnn) v8.5 or above
197 - [Compiler](https://gist.github.com/ax3l/9489132) compatible with CUDA
199 ….com/deeplearning/cudnn/reference/support-matrix.html) for cuDNN versions with the various support…
201 If you want to disable CUDA support, export the environment variable `USE_CUDA=0`.
204 …ilable here](https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/)
208 - [AMD ROCm](https://rocm.docs.amd.com/en/latest/deploy/linux/quick_start.html) 4.0 and above insta…
209 - ROCm is currently supported only for Linux systems.
216 - [PyTorch Prerequisites for Intel GPUs](https://www.intel.com/content/www/us/en/developer/articles…
217 - Intel GPU is supported for Linux and Windows.
224 git clone --recursive https://github.com/pytorch/pytorch
228 git submodule update --init --recursive
240 pip install -r requirements.txt
246 pip install mkl-static mkl-include
247 # CUDA only: Add LAPACK support for the GPU if needed
248 conda install -c pytorch magma-cuda121 # or the magma-cuda* that matches your CUDA version from ht…
260 pip install mkl-static mkl-include
262 conda install pkg-config libuv
268 pip install mkl-static mkl-include
271 conda install -c conda-forge libuv=1.39
282 …build with XPU supports both new and old C++ ABIs. Previously, XPU only supported the new C++ ABI.…
292 export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
296 > _Aside:_ If you are using [Anaconda](https://www.anaconda.com/distribution/#download-section), yo…
299 > build/temp.linux-x86_64-3.7/torch/csrc/stub.o: file not recognized: file format not recognized
314 …build legacy python code, please refer to [Building on legacy code and CUDA](https://github.com/py…
316 **CPU-only builds**
324 …thub.com/pytorch/pytorch/blob/main/docs/source/notes/windows.rst#building-from-source) is an examp…
326 **CUDA based build**
328 In this mode PyTorch computations will leverage your GPU via CUDA for faster number crunching
330 …ameworkslibrary/nvtx/nvidia_tools_extension_library_nvtx.htm) is needed to build Pytorch with CUDA.
331 NVTX is a part of CUDA distributive, where it is called "Nsight Compute". To install it onto an alr…
332 Make sure that CUDA with Nsight Compute is installed after Visual Studio.
338 …-src/oneDNN), and [Sccache](https://github.com/mozilla/sccache) are often needed. Please refer to …
340 …pytorch.bat](https://github.com/pytorch/pytorch/blob/main/.ci/pytorch/win-test-helpers/build_pytor…
352 …t to override the underlying toolset used by Ninja and Visual Studio with CUDA, please run the fol…
357 …ler\vswhere.exe" -version [15^,17^) -products * -latest -property installationPath`) do call "%i\V…
359 :: [Optional] If you want to override the CUDA host compiler
366 ##### Adjust Build Options (Optional)
369 the following. For example, adjusting the pre-detected directories for CuDNN or BLAS can be done
374 export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
375 python setup.py build --cmake-only
376 ccmake build # or cmake-gui build
381 export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
382 MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build --cmake-only
383 ccmake build # or cmake-gui build
386 ### Docker Image
388 #### Using pre-built images
390 You can also pull a pre-built docker image from Docker Hub and run with docker v19.03+
393 docker run --gpus all --rm -ti --ipc=host pytorch/pytorch:latest
398 …d increase shared memory size either with `--ipc=host` or `--shm-size` command line options to `nv…
402 **NOTE:** Must be built with a docker version > 18.06
404 The `Dockerfile` is supplied to build images with CUDA 11.1 support and cuDNN v8.
409 make -f docker.Makefile
410 # images are tagged as docker.io/${your_docker_username}/pytorch
413 … environment variable to specify additional CMake variables to be passed to CMake during the build.
417 make -f docker.Makefile
422 To build documentation in various formats, you will need [Sphinx](http://www.sphinx-doc.org) and the
427 pip install -r requirements.txt
429 You can then build the documentation by running `make <format>` from the
433 `npm install -g katex`
440 ```npm install -g katex@0.13.18```
445 on [our website](https://pytorch.org/previous-versions).
450 Three-pointers to get you started:
451 - [Tutorials: get you started with understanding and using PyTorch](https://pytorch.org/tutorials/)
452 - [Examples: easy to understand PyTorch code across all domains](https://github.com/pytorch/example…
453 - [The API Reference](https://pytorch.org/docs/)
454 - [Glossary](https://github.com/pytorch/pytorch/blob/main/GLOSSARY.md)
462 …ep Learning with PyTorch from Udacity](https://www.udacity.com/course/deep-learning-pytorch--ud188)
463 …h PyTorch from Udacity](https://www.udacity.com/course/intro-to-machine-learning-nanodegree--nd229)
464 …works with PyTorch from Coursera](https://www.coursera.org/learn/deep-neural-networks-with-pytorch)
473 * Newsletter: No-noise, a one-way email newsletter with important announcements about PyTorch. You …
481 We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so wit…
490 PyTorch is a community-driven project with several skillful engineers and researchers contributing …
493 …-exhaustive but growing list needs to mention: [Trevor Killeen](https://github.com/killeent), [Sas…
499 PyTorch has a BSD-style license, as found in the [LICENSE](LICENSE) file.